Bump fallback version to v1.5 and upgrade dependencies

Author: amaslenn

pyproject.toml

@@ -18,42 +18,42 @@
 name = "cloudai"
 dynamic = ["version"]
 dependencies = [
-  "bokeh==3.4.1",
-  "pandas==2.2.1",
-  "tbparse==0.0.8",
-  "toml==0.10.2",
-  "kubernetes==30.1.0",
-  "pydantic==2.8.2",
-  "jinja2==3.1.6",
-  "websockets==15.0.1",
-  "rich==14.0.0",
-  "click>=8.2.1",
+  "bokeh~=3.8",
+  "pandas~=2.3",
+  "tbparse~=0.0.9",
+  "toml~=0.10.2",
+  "kubernetes~=34.1",
+  "pydantic~=2.12",
+  "jinja2~=3.1.6",
+  "websockets~=15.0.1",
+  "rich~=14.2",
+  "click~=8.3",
 ]
 requires-python = ">=3.10"
 scripts = { cloudai = "cloudai.cli:main" }
 
   [project.optional-dependencies]
   dev = [
-    "ruff==0.11.7",
-    "pandas-stubs==2.2.*",
-    "pyright==1.1.400",
-    "build==1.2.*",
+    "ruff~=0.14",
+    "pandas-stubs~=2.3",
+    "pyright~=1.1",
+    "build~=1.3",
     "vulture==2.14",
-    "pytest==8.3.*",
-    "pytest-cov==6.1.*",
-    "pytest-asyncio==0.26.*",
-    "import-linter==2.3.*",
-    "pytest-deadfixtures==2.2.*",
-    "taplo==0.9.3",
+    "pytest~=8.4",
+    "pytest-cov~=7.0",
+    "pytest-asyncio~=1.2",
+    "import-linter~=2.5",
+    "pytest-deadfixtures~=2.2",
+    "taplo~=0.9.3",
   ]
   docs = [
-    "sphinx>=7.0",
-    "nvidia-sphinx-theme",
-    "myst-parser",
-    "sphinxext-opengraph",
-    "sphinx-design",
-    "sphinxcontrib-mermaid",
-    "sphinx-copybutton>=0.5.2",
+    "sphinx~=8.1",
+    "nvidia-sphinx-theme~=0.0.8",
+    "myst-parser~=4.0",
+    "sphinxext-opengraph~=0.13",
+    "sphinx-design~=0.6",
+    "sphinxcontrib-mermaid~=1.0",
+    "sphinx-copybutton~=0.5",
   ]
 
 [build-system]
@@ -62,7 +62,7 @@ build-backend = "pdm.backend"
 
 [tool.pdm.version]
 source = "scm"
-fallback_version = "1.4.0+unknown"
+fallback_version = "1.5.0+unknown"
 
 [tool.ruff]
 line-length = 120

src/cloudai/cli/handlers.py

@@ -149,7 +149,7 @@ def handle_dse_job(runner: Runner, args: argparse.Namespace) -> int:
             step, action = result
             env.test_run.step = step
             logging.info(f"Running step {step} (of {agent.max_steps}) with action {action}")
-            observation, reward, done, info = env.step(action)
+            observation, reward, *_ = env.step(action)
             feedback = {"trial_index": step, "value": reward}
             agent.update_policy(feedback)
             logging.info(f"Step {step}: Observation: {[round(obs, 4) for obs in observation]}, Reward: {reward:.4f}")

src/cloudai/report_generator/tool/bokeh_report_tool.py

@@ -18,7 +18,7 @@
 
 from math import pi
 from pathlib import Path
-from typing import TYPE_CHECKING, List, Optional, Tuple
+from typing import TYPE_CHECKING, List, Literal, Optional, Tuple
 
 from cloudai.report_generator.util import bokeh_size_unit_js_tick_formatter, calculate_power_of_two_ticks
 from cloudai.util.lazy_imports import lazy
@@ -50,7 +50,7 @@ def create_figure(
         y_range: bokeh.models.Range1d,
         width: int = 500,
         height: int = 308,
-        x_axis_type: str = "linear",
+        x_axis_type: Literal["linear", "log"] = "linear",
         tools: str = "pan,wheel_zoom,box_zoom,reset,save",
         x_range: Optional[bokeh.models.Range1d] = None,
     ) -> bokeh.plotting.figure:

src/cloudai/systems/lsf/lsf_system.py

@@ -168,7 +168,7 @@ def parse_bjobs_output(self, bjobs_output: str) -> Dict[str, str]:
             parts = line.split()
             if len(parts) < 6:
                 continue
-            job_id, user, _, _, _, exec_host = parts[:6]
+            _, user, _, _, _, exec_host = parts[:6]
             if exec_host not in node_user_map:
                 node_user_map[exec_host] = user
         return node_user_map

src/cloudai/workloads/chakra_replay/report_generation_strategy.py

@@ -225,7 +225,7 @@ def _generate_bokeh_content(
             toolbar_location="right",
             tools="hover",
             tooltips="@comm: @value (@percentage)",
-            x_range=(-0.5, 1.5),
+            x_range=lazy.bokeh_models.Range(-0.5, 1.5),
         )
         p.add_layout(lazy.bokeh_models.Title(text="Comm Types Distribution", align="center"), "above")
         p.wedge(

src/cloudai/workloads/nccl_test/prediction_report_generator.py

@@ -117,7 +117,7 @@ def _run_predictor(self, gpu_type: str) -> pd.DataFrame:
         if predictor_paths is None:
             return lazy.pd.DataFrame()
 
-        config_path, model_path, input_csv, output_csv, predictor_sub_path = predictor_paths
+        config_path, model_path, input_csv, output_csv, _ = predictor_paths
 
         if not self.predictor or not self.predictor.venv_path:
             logging.warning("Predictor virtual environment is not set up. Skipping prediction.")

tests/slurm_command_gen_strategy/test_nemo_launcher_slurm_command_gen_strategy.py

@@ -201,7 +201,7 @@ def test_data_prefix_validation(self, cmd_gen_strategy: NeMoLauncherSlurmCommand
         tdef.cmd_args.training.model.data.data_impl = "not_mock"
         tdef.cmd_args.training.model.data.data_prefix = "[]"
 
-        with pytest.raises(ValueError, match="The 'data_prefix' field of the NeMo launcher test is missing or empty."):
+        with pytest.raises(ValueError, match=r"The 'data_prefix' field of the NeMo launcher test is missing or empty."):
             cmd_gen_strategy.gen_exec_command()
 
     @patch("pathlib.Path.open", new_callable=mock_open)

tests/test_cloudaigym.py

@@ -281,7 +281,9 @@ def test_params_set_validated(setup_env: tuple[TestRun, Runner], nemorun: NeMoRu
 
     assert excinfo.type is UserWarning
     assert "Pydantic serializer warnings:" in str(excinfo.value)
-    assert "but got `str`" in str(excinfo.value)
+    assert "serialized value may not be as expected [field_name='max_steps', input_value='invalid'" in str(
+        excinfo.value
+    )
 
 
 def test_apply_params_set__preserves_installables_state(setup_env: tuple[TestRun, Runner], tmp_path: Path):

tests/test_csv_report_tool.py

@@ -42,7 +42,7 @@ def test_set_dataframe(example_dataframe: pd.DataFrame):
 
 def test_finalize_report_no_dataframe(output_directory: Path):
     csv_tool = CSVReportTool(output_directory=output_directory)
-    with pytest.raises(ValueError, match="No DataFrame has been set for the report."):
+    with pytest.raises(ValueError, match=r"No DataFrame has been set for the report."):
         csv_tool.finalize_report(Path("report.csv"))
 
 

tests/test_slurm_installer.py

@@ -278,7 +278,7 @@ def test_install_python_executable_prefers_pyproject_toml(
     def test_install_python_executable_prefers_requirements_txt(
         self, installer: SlurmInstaller, git: GitRepo, setup_repo
     ):
-        repo_dir, subdir, _, requirements_file = setup_repo
+        repo_dir, *_ = setup_repo
 
         py = PythonExecutable(git, project_subpath=Path("subdir"), dependencies_from_pyproject=False)