konfai 1.1.8__tar.gz → 1.2.0__tar.gz

{konfai-1.1.8 → konfai-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konfai
-Version: 1.1.8
+Version: 1.2.0
 Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
 Author-email: Valentin Boussot <boussot.v@gmail.com>
 License-Expression: Apache-2.0
@@ -27,8 +27,6 @@ Provides-Extra: lpips
 Requires-Dist: lpips; extra == "lpips"
 Provides-Extra: cluster
 Requires-Dist: submitit; extra == "cluster"
-Provides-Extra: plot
-Requires-Dist: matplotlib; extra == "plot"
 Dynamic: license-file
 
 [![License](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://github.com/vboussot/KonfAI/blob/main/LICENSE)

konfai-1.2.0/konfai/__init__.py

@@ -0,0 +1,61 @@
+import datetime
+import os
+
+
+def models_directory() -> str:
+    return _get_env("KONFAI_MODELS_DIRECTORY")
+
+
+def checkpoints_directory() -> str:
+    return _get_env("KONFAI_CHECKPOINTS_DIRECTORY")
+
+
+def path_to_models() -> str:
+    return _get_env("KONFAI_MODEL")
+
+
+def predictions_directory() -> str:
+    return _get_env("KONFAI_PREDICTIONS_DIRECTORY")
+
+
+def evaluations_directory() -> str:
+    return _get_env("KONFAI_EVALUATIONS_DIRECTORY")
+
+
+def statistics_directory() -> str:
+    return _get_env("KONFAI_STATISTICS_DIRECTORY")
+
+
+def setups_directory() -> str:
+    return _get_env("KONFAI_SETUPS_DIRECTORY")
+
+
+def config_file() -> str:
+    return _get_env("KONFAI_config_file")
+
+
+def konfai_state() -> str:
+    return _get_env("KONFAI_STATE")
+
+
+def konfai_root() -> str:
+    return _get_env("konfai_root")
+
+
+def cuda_visible_devices() -> str:
+    return _get_env("CUDA_VISIBLE_DEVICES")
+
+
+def konfai_nb_cores() -> str:
+    return _get_env("KONFAI_NB_CORES")
+
+
+def current_date() -> str:
+    return datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
+
+
+def _get_env(var: str) -> str:
+    value = os.environ.get(var)
+    if value is None:
+        raise RuntimeError(f"Environment variable '{var}' is not set.")
+    return value