Remove unused context imports from test files - all Ruff errors fixed

Co-authored-by: njzjz <[email protected]>

Author: Copilot

tests/exploration/test_exploration_scheduler.py

@@ -2,13 +2,7 @@
 
 import numpy as np
 
-try:
-    from exploration.context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from dflow.python import (
     FatalError,
 )

tests/exploration/test_lmp_templ_task_group.py

@@ -7,13 +7,7 @@
 )
 
 
-try:
-    from exploration.context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 
 from dpgen2.constants import (
     lmp_conf_name,

tests/exploration/test_make_task_group_from_config.py

@@ -5,13 +5,7 @@
 )
 
 
-try:
-    from exploration.context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from dpgen2.exploration.task import (
     CalyTaskGroup,
     LmpTemplateTaskGroup,

tests/mocked_ops.py

@@ -20,13 +20,7 @@
     Tuple,
 )
 
-try:
-    from flow.context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from dpgen2.constants import (
     fp_task_pattern,
     lmp_conf_name,

tests/test_block_cl.py

@@ -13,13 +13,7 @@
     upload_artifact,
 )
 
-try:
-    from context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from context import (
     default_host,
     default_image,

tests/test_caly_evo_step.py

@@ -23,13 +23,7 @@
     calypso_run_opt_file,
 )
 
-try:
-    from context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from context import (
     default_host,
     default_image,

tests/test_collect_data.py

@@ -16,13 +16,7 @@
     PythonOPTemplate,
 )
 
-try:
-    from context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from context import (
     default_host,
     default_image,

tests/test_dpgen_loop.py

@@ -13,13 +13,7 @@
     upload_artifact,
 )
 
-try:
-    from context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from context import (
     default_image,
     skip_ut_with_dflow,

tests/test_merge_caly_evo_step.py

@@ -24,13 +24,7 @@
     calypso_run_opt_file,
 )
 
-try:
-    from context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from context import (
     default_host,
     default_image,

tests/test_prep_run_caly.py

@@ -17,13 +17,7 @@
     calypso_run_opt_file,
 )
 
-try:
-    from context import (
-        dpgen2,
-    )
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
+# case of upload everything to argo, no context needed
 from context import (
     default_host,
     default_image,