[RLlib] Fix broken tune tests in master due to framework=auto errors. (#8672)

python/ray/tune/tests/test_cluster.py

@@ -554,7 +554,7 @@ ray.init(address="{address}")
 tune.run(
     "PG",
     name="experiment",
-    config=dict(env="CartPole-v1"),
+    config=dict(env="CartPole-v1", framework="tf"),
     stop=dict(training_iteration=10),
     local_dir="{checkpoint_dir}",
     checkpoint_freq=1,
@@ -593,7 +593,7 @@ tune.run(
             "experiment": {
                 "run": "PG",
                 "checkpoint_freq": 1,
-                "local_dir": dirpath
+                "local_dir": dirpath,
             }
         },
         resume=True)

python/ray/tune/tests/test_tune_restore.py

@@ -36,6 +36,7 @@ class TuneRestoreTest(unittest.TestCase):
             local_dir=tmpdir,
             config={
                 "env": "CartPole-v0",
+                "framework": "tf",
             },
         )
 
@@ -58,6 +59,7 @@ class TuneRestoreTest(unittest.TestCase):
             restore=self.checkpoint_path,  # Restore the checkpoint
             config={
                 "env": "CartPole-v0",
+                "framework": "tf",
             },
         )
 
@@ -73,6 +75,7 @@ class TuneRestoreTest(unittest.TestCase):
             restore=self.checkpoint_path,
             config={
                 "env": "CartPole-v0",
+                "framework": "tf",
             },
         )
         self.assertTrue(os.path.isfile(self.checkpoint_path))