chore: remove wandb_init from sweep tests (#9008)

Co-authored-by: Dmitry Duev <dmitryduev@users.noreply.github.com>

tests/system_tests/test_sweep/test_sweep_public_api.py

@@ -96,8 +96,8 @@ def test_from_path(user):
     assert isinstance(sweep, wandb.apis.public.Sweep)
 
 
-def test_project_sweeps(user, wandb_init):
-    run = wandb_init(entity=user, project="testnosweeps")
+def test_project_sweeps(user):
+    run = wandb.init(entity=user, project="testnosweeps")
     run.finish()
     sweep_id = wandb.sweep(SWEEP_CONFIG_BAYES, entity=user, project="test")
     api = Api()

tests/system_tests/test_sweep/test_sweep_scheduler.py

@@ -182,12 +182,12 @@ def mock_run_complete_scheduler(self, *args, **kwargs):
     assert scheduler.state == SchedulerState.FAILED
 
 
-def test_sweep_scheduler_sweep_id_with_job(user, wandb_init, monkeypatch):
+def test_sweep_scheduler_sweep_id_with_job(user, monkeypatch):
     _patch_wandb_run(monkeypatch)
     sweep_config = VALID_SWEEP_CONFIGS_MINIMAL[0]
 
     # make a job
-    run = wandb_init()
+    run = wandb.init()
     job_artifact = run._log_job_artifact_with_image("ljadnfakehbbr", args=[])
     job_name = job_artifact.wait().name
     sweep_config["job"] = job_name
@@ -626,13 +626,11 @@ def mock_stop_run(*args, **kwargs):
     assert "mock-run-id-1" not in _scheduler._runs
 
 
-def test_launch_sweep_scheduler_try_executable_works(
-    user, wandb_init, test_settings, monkeypatch
-):
+def test_launch_sweep_scheduler_try_executable_works(user, test_settings, monkeypatch):
     _patch_wandb_run(monkeypatch)
     _project = "test-project"
     settings = test_settings({"project": _project})
-    run = wandb_init(settings=settings)
+    run = wandb.init(settings=settings)
     job_artifact = run._log_job_artifact_with_image("lala-docker-123", args=[])
     job_name = f"{user}/{_project}/{job_artifact.wait().name}"
 

tests/system_tests/test_sweep/test_wandb_agent_full.py

@@ -7,7 +7,7 @@
 from wandb.apis.public import Api
 
 
-def test_agent_basic(wandb_init):
+def test_agent_basic(user):
     sweep_ids = []
     sweep_configs = []
     sweep_resumed = []
@@ -19,7 +19,7 @@ def test_agent_basic(wandb_init):
     }
 
     def train():
-        run = wandb_init()
+        run = wandb.init()
         sweep_ids.append(run.sweep_id)
         sweep_configs.append(dict(run.config))
         sweep_resumed.append(run.resumed)
@@ -35,11 +35,11 @@ def train():
     assert sweep_resumed[0] is False
 
 
-def test_agent_config_merge(wandb_init):
+def test_agent_config_merge(user):
     sweep_configs = []
 
     def train():
-        run = wandb_init(config={"extra": 2})
+        run = wandb.init(config={"extra": 2})
         sweep_configs.append(dict(run.config))
         run.finish()
 
@@ -57,11 +57,11 @@ def train():
     assert sweep_configs[0] == {"a": 1, "extra": 2}
 
 
-def test_agent_config_ignore(wandb_init):
+def test_agent_config_ignore(user):
     sweep_configs = []
 
     def train():
-        run = wandb_init(config={"a": "ignored", "extra": 2})
+        run = wandb.init(config={"a": "ignored", "extra": 2})
         sweep_configs.append(dict(run.config))
         run.finish()
 

tests/unit_tests/test_data_types.py

@@ -395,27 +395,6 @@ def test_fail_to_make_file(
         wb_image.bind_to_run(mock_run(), "my key: an identifier", 0)
 
 
-# def test_cant_serialize_to_other_run(mock_run):
-#     """This isn't implemented yet. Should work eventually."""
-#     other_run = wandb.wandb_sdk.wandb_run.Run(settings=wandb_init)
-#     other_run._set_backend(mock_run._backend)
-#     wb_image = wandb.Image(image)
-
-#     wb_image.bind_to_run(mock_run, "stuff", 10)
-
-#     with pytest.raises(AssertionError):
-#         wb_image.to_json(other_run)
-
-
-#     meta_expected = {
-#         "_type": "images/separated",
-#         "count": 1,
-#         "height": 28,
-#         "width": 28,
-#     }
-#     assert utils.subdict(meta, meta_expected) == meta_expected
-
-
 ################################################################################
 # Test wandb.Audio
 ################################################################################