libinephany 1.0.3__py3-none-any.whl → 1.1.1__py3-none-any.whl

libinephany/observations/observers/local_observers.py

@@ -683,6 +683,14 @@ class ModuleTypeOneHot(LocalObserver):
 
         return False
 
+    @property
+    def can_standardize(self) -> bool:
+        """
+        :return: Whether the observation can be standardized.
+        """
+
+        return False
+
     def _get_observation_format(self) -> StatisticStorageTypes:
         """
         :return: Format the observation returns data in. Must be one of the enum attributes in the StatisticStorageTypes

libinephany/pydantic_models/schemas/inner_task_profile.py

@@ -27,6 +27,7 @@ class InnerTaskProfile(BaseModel):
     number_of_parameters: int
     vram_usage: float
     idle_vram_usage: float
+    hparam_overrides: dict[str, dict[str, Any]] | None = None
 
     @field_validator("vram_usage", "idle_vram_usage", mode="before")
     def replace_non_with_nan(cls, value: None | float) -> float:
@@ -244,6 +245,7 @@ class InnerTaskProfiles(BaseModel):
         number_of_parameters: int,
         vram_usage: float,
         idle_vram_usage: float,
+        hparam_overrides: dict[str, dict[str, Any]] | None = None,
     ) -> None:
         """
         :param inner_task_name: Name of the inner task to add a profile for.
@@ -254,6 +256,7 @@ class InnerTaskProfiles(BaseModel):
         :param vram_usage: VRAM required to perform the inner task. Can be NaN if an OOM was encountered.
         :param idle_vram_usage: VRAM required for the inner task to sit loaded but not actively being trained. Can be
         NaN if an OOM was encountered.
+        :param hparam_overrides: Hyperparameter overrides for the inner task.
         """
 
         self.profiles[inner_task_name] = InnerTaskProfile(
@@ -265,6 +268,7 @@ class InnerTaskProfiles(BaseModel):
             number_of_parameters=number_of_parameters,
             vram_usage=vram_usage,
             idle_vram_usage=idle_vram_usage,
+            hparam_overrides=hparam_overrides,
         )
 
     def validate_task_profiles(self, policy_mapping_function: Callable[[str, Any, Any], str]) -> None:

libinephany/pydantic_models/schemas/tensor_statistics.py

@@ -205,7 +205,7 @@ class TensorStatistics(BaseModel):
         for field, field_value in stats.model_dump().items():
             name = field[:-1] if field.endswith(FIELD_SUFFIX) else field
 
-            if name in include_statistics:
+            if name in include_statistics or field in include_statistics:
                 setattr(stats, name, downsampled_tensor)
 
         return stats

libinephany/pydantic_models/states/hyperparameter_states.py

@@ -60,9 +60,7 @@ class UpdateCallbacks(BaseModel):
     adam_eps: Callable[..., None]
     sgd_momentum: Callable[..., None]
 
-    batch_size: Callable[..., None] | None
     gradient_accumulation: Callable[..., None] | None
-    epochs: Callable[..., None] | None
 
     def __getitem__(self, item: str) -> Callable[..., None] | None:
         """

{libinephany-1.0.3.dist-info → libinephany-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: libinephany
-Version: 1.0.3
+Version: 1.1.1
 Summary: Inephany library containing code commonly used by multiple subpackages.
 Author-email: Inephany <info@inephany.com>
 License: Apache 2.0

{libinephany-1.0.3.dist-info → libinephany-1.1.1.dist-info}/RECORD

@@ -9,7 +9,7 @@ libinephany/observations/statistic_manager.py,sha256=LLg1zSxnJr2oQQepYla3qoUuRy1
 libinephany/observations/statistic_trackers.py,sha256=F98V-H2Ljx0v2YnppYCCJLJojL6pzYBdbBh8Lb4lasA,47666
 libinephany/observations/observers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/observations/observers/base_observers.py,sha256=V8PIysq2wT6K-w_CqeM5benyif-xK1hPT3M6a4ic1So,17535
-libinephany/observations/observers/local_observers.py,sha256=NVH7fIaV2rIIvXvyntlwOHaKhGyxj_zenqG-4SOAeNM,46156
+libinephany/observations/observers/local_observers.py,sha256=HvOQhVEANQJjAl6Ln_YuP0-fNt_Xkh8ixlkb-A5JqDM,46318
 libinephany/observations/observers/observer_containers.py,sha256=VNyqGgxYJ4r49Msp_kk-POgicb-_5w54twuT1qfNdxw,9562
 libinephany/observations/observers/global_observers/__init__.py,sha256=87WHRPYmL0tVsaTKUd91pwEpCZtHPSKRQoba2VQjswA,3018
 libinephany/observations/observers/global_observers/base_classes.py,sha256=Q7OblhmKscypTs9JBepSQwo6ljjOdPKTU9kbpuhq_W4,7800
@@ -28,13 +28,13 @@ libinephany/pydantic_models/configs/observer_config.py,sha256=v_ChzaVXC_rlZ7eDZP
 libinephany/pydantic_models/configs/outer_model_config.py,sha256=GQ0QBSC2Xht8x8X_TEMfYM2GF_x1kErLuFrA_H6Jhs0,1209
 libinephany/pydantic_models/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/pydantic_models/schemas/agent_info.py,sha256=me5gDxvZjP9TNK588mpUvxiiJrPDqy3Z7ZHRzryAYTs,2628
-libinephany/pydantic_models/schemas/inner_task_profile.py,sha256=Xu0tQmhGwV043tTamFiHekuE1RRXhhrUrGbtymjXo7g,11722
+libinephany/pydantic_models/schemas/inner_task_profile.py,sha256=2ix497uUCHUJzdfXtJRT6MDDn8OFQZ8zpZMBZUlvIHc,11976
 libinephany/pydantic_models/schemas/observation_models.py,sha256=MLhxqDet9Yol1D5mkQGQsQT23sm37AStRLnPc4sgcZc,2110
 libinephany/pydantic_models/schemas/request_schemas.py,sha256=VED8eAUvBofxeAx9gWU8DyCZOTVD3QsHRq-TO7kyOqk,1260
 libinephany/pydantic_models/schemas/response_schemas.py,sha256=SKFuasdjX5aH_I0vT3SwnpwhyMf9cNPB1ZpDeAGgoO8,2158
-libinephany/pydantic_models/schemas/tensor_statistics.py,sha256=hWl52SxPYHFBXUwwZ9X7E1iFhuxGsPypd14hFTx91jw,8166
+libinephany/pydantic_models/schemas/tensor_statistics.py,sha256=BTgeLacmhqc3uUM0WJiJdTwmdQe2RSKR9f4o5D8FZ1Y,8197
 libinephany/pydantic_models/states/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-libinephany/pydantic_models/states/hyperparameter_states.py,sha256=Esi1xdrH9xOJwhpSezkfbTzbKX4O26IpN5zzkWD3Mf8,33779
+libinephany/pydantic_models/states/hyperparameter_states.py,sha256=SCLj-e84AsvGx2BIsl1moNQAQ79HU4Yqk1-xp3rr8sY,33697
 libinephany/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/utils/agent_utils.py,sha256=_2w1AY5Y4mQ5hes_Rq014VhZXOtIOn-W92mZgeixv3g,2658
 libinephany/utils/asyncio_worker.py,sha256=Ew23zKIbG1zwyCudcyiObMrw4G0f3p2QXzZfM4mePqI,2751
@@ -57,8 +57,8 @@ libinephany/utils/typing.py,sha256=rGbaPO3MaUndsWiC_wHzReD_TOLYqb43i01pKN-j7Xs,6
 libinephany/web_apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/web_apps/error_logger.py,sha256=gAQIaqerqP4ornXZwFF1cghjnd2mMZEt3aVrTuUCr34,16653
 libinephany/web_apps/web_app_utils.py,sha256=qiq_lasPipgN1RgRudPJc342kYci8O_4RqppxmIX8NY,4095
-libinephany-1.0.3.dist-info/licenses/LICENSE,sha256=pogfDoMBP07ehIOvWymuWIar8pg2YLUhqOHsJQU3wdc,9250
-libinephany-1.0.3.dist-info/METADATA,sha256=AJHvThkrO8rYniWmXEWyyH8hnKNMIllOu7VGvCbeTo0,8389
-libinephany-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libinephany-1.0.3.dist-info/top_level.txt,sha256=bYAOXQdJgIoLkO2Ui0kxe7pSYegS_e38u0dMscd7COQ,12
-libinephany-1.0.3.dist-info/RECORD,,
+libinephany-1.1.1.dist-info/licenses/LICENSE,sha256=pogfDoMBP07ehIOvWymuWIar8pg2YLUhqOHsJQU3wdc,9250
+libinephany-1.1.1.dist-info/METADATA,sha256=Tuhz8AcpTpOjIu4XuUd7dJsI8874RRP3RWn4OlpK-Sc,8389
+libinephany-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libinephany-1.1.1.dist-info/top_level.txt,sha256=bYAOXQdJgIoLkO2Ui0kxe7pSYegS_e38u0dMscd7COQ,12
+libinephany-1.1.1.dist-info/RECORD,,