Manage plot limit

pola-rs · Apr 18, 2024 · 70f1e06 · 70f1e06
1 parent 3b11de5
commit 70f1e06
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/scripts/plot_bars.py b/scripts/plot_bars.py
@@ -50,7 +50,7 @@
     "csv": 25.0,
     "feather": 20.0,
 }
-LIMIT = Y_LIMIT_MAP[settings.run.io_type]
+LIMIT = settings.plot.y_limit or Y_LIMIT_MAP[settings.run.io_type]
 
 
 def main() -> None:

diff --git a/settings.py b/settings.py
@@ -51,8 +51,7 @@ def include_io(self) -> bool:
 class Plot(BaseSettings):
     show: bool = False
     n_queries: int = 7
-    limit_with_io: float = 20.0
-    limit_without_io: float = 15.0
+    y_limit: float | None = None
 
     model_config = SettingsConfigDict(
         env_prefix="plot_", env_file=".env", extra="ignore"