Pydantic ignore

ml-energy · Feb 26, 2025 · 72af888 · 72af888
1 parent 089fd54
commit 72af888
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/zeus/optimizer/pipeline_frequency/common.py b/zeus/optimizer/pipeline_frequency/common.py
@@ -64,7 +64,7 @@ def _validate_scheduler_args(cls, args, values):
     def _make_upper_case(cls, value):
         return value.upper()
 
-    class PFOConfig:  # type: ignore
+    class Config:  # type: ignore
         """Configuration class read by pydantic."""
 
         env_prefix = "zeus_pfo_"

diff --git a/zeus/optimizer/pipeline_frequency/profiling/models.py b/zeus/optimizer/pipeline_frequency/profiling/models.py
@@ -72,7 +72,7 @@ def _set_mode(self, mode, values):
 
         return scheduler.mode
 
-    class Config:
+    class PerseusConfig:  # type: ignore
         """Configuration class read by pydantic."""
 
         env_prefix = "perseus_"