libinephany 0.13.8__py3-none-any.whl → 0.14.0__py3-none-any.whl

libinephany/pydantic_models/configs/hyperparameter_configs.py

@@ -293,6 +293,10 @@ class HParamConfigs(BaseModel):
     token_config: HParamConfig = TokensHParamConfig()
     samples_config: HParamConfig = SamplesHParamConfig()
 
+    constraints: dict[str, dict[str, Any] | None] | None = {
+        "FirstDefaultParamorphConstraint": None,
+    }
+
     def override(self, overrides: dict[str, dict[str, Any]] | None) -> None:
         """
         :param overrides: None or config overrides to apply.

libinephany/pydantic_models/schemas/response_schemas.py

@@ -37,6 +37,24 @@ class ClientScheduleResponse(BaseModel):
 
     response_time: float | None = None
 
+    def get_value_mapping(self) -> dict[str, float | int]:
+        """
+        :return: Dictionary mapping agent IDs to their values.
+        """
+
+        return {
+            agent_id: agent_response.hyperparameter_internal_value for agent_id, agent_response in self.actions.items()
+        }
+
+    def set_values_from_mapping(self, values: dict[str, float | int]) -> None:
+        """
+        :param values: Dictionary mapping agent IDs to their values.
+        """
+
+        for agent_id, value in values.items():
+            if agent_id in self.actions:
+                self.actions[agent_id].hyperparameter_internal_value = value
+
 
 class ClientPolicySchemaResponse(BaseModel):
     observation_clipping_threshold: float

libinephany/pydantic_models/states/hyperparameter_states.py

@@ -309,6 +309,56 @@ class HyperparameterContainer(BaseModel):
             if isinstance(field_value, Hyperparameter) and field_name not in skip_hparams
         }
 
+    def get_current_internal_values(self, skip_hparams: list[str] | None = None) -> dict[str, float | int | None]:
+        """
+        :param skip_hparams: Names of the hyperparameters to not include in the current values vector returned by
+        this observation.
+        :return: Dictionary mapping hyperparameter names to their current values for this parameter group.
+        """
+
+        if skip_hparams is None:
+            skip_hparams = []
+
+        return {
+            field_name: field_value.current_internal_value
+            for field_name, field_value in self.__dict__.items()
+            if isinstance(field_value, Hyperparameter) and field_name not in skip_hparams
+        }
+
+    def get_current_deltas(self, skip_hparams: list[str] | None = None) -> dict[str, float | int | None]:
+        """
+        :param skip_hparams: Names of the hyperparameters to not include in the current deltas vector returned by
+        this observation.
+        :return: Dictionary mapping hyperparameter names to their current deltas for this parameter group.
+        """
+
+        if skip_hparams is None:
+            skip_hparams = []
+
+        return {
+            field_name: field_value.current_delta
+            for field_name, field_value in self.__dict__.items()
+            if isinstance(field_value, Hyperparameter) and field_name not in skip_hparams
+        }
+
+    def set_internal_values(self, internal_values: dict[str, float | int | None]) -> None:
+        """
+        :param internal_values: Dictionary mapping hyperparameter names to their new internal values.
+        """
+
+        for field_name, field_value in internal_values.items():
+            if hasattr(self, field_name) and isinstance(getattr(self, field_name), Hyperparameter):
+                getattr(self, field_name).internal_value = field_value
+
+    def set_deltas(self, deltas: dict[str, float | int | None]) -> None:
+        """
+        :param deltas: Dictionary mapping hyperparameter names to their new deltas.
+        """
+
+        for field_name, field_value in deltas.items():
+            if hasattr(self, field_name) and isinstance(getattr(self, field_name), Hyperparameter):
+                getattr(self, field_name).delta = field_value
+
     def set_sampled_initial_values(
         self, sample_initial_external_values: dict[str, np.ndarray | float | int | None]
     ) -> None:
@@ -382,39 +432,10 @@ class ParameterGroupHParams(HyperparameterContainer):
                 name=hyperparameter_name,
             )
 
-        return cls(parameter_group_name=parameter_group_name, **hyperparameters)
-
-    def get_current_internal_values(self, skip_hparams: list[str] | None = None) -> dict[str, float | int | None]:
-        """
-        :param skip_hparams: Names of the hyperparameters to not include in the current values vector returned by
-        this observation.
-        :return: Dictionary mapping hyperparameter names to their current values for this parameter group.
-        """
-
-        if skip_hparams is None:
-            skip_hparams = []
-
-        return {
-            field_name: field_value.current_internal_value
-            for field_name, field_value in self.__dict__.items()
-            if isinstance(field_value, Hyperparameter) and field_name not in skip_hparams
-        }
-
-    def get_current_deltas(self, skip_hparams: list[str] | None = None) -> dict[str, float | int | None]:
-        """
-        :param skip_hparams: Names of the hyperparameters to not include in the current deltas vector returned by
-        this observation.
-        :return: Dictionary mapping hyperparameter names to their current deltas for this parameter group.
-        """
-
-        if skip_hparams is None:
-            skip_hparams = []
-
-        return {
-            field_name: field_value.current_delta
-            for field_name, field_value in self.__dict__.items()
-            if isinstance(field_value, Hyperparameter) and field_name not in skip_hparams
-        }
+        return cls(
+            parameter_group_name=parameter_group_name,
+            **hyperparameters,
+        )
 
     def get_hyperparameter_transform_types(
         self, skip_hparams: list[str] | None = None

{libinephany-0.13.8.dist-info → libinephany-0.14.0.dist-info}/METADATA

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

{libinephany-0.13.8.dist-info → libinephany-0.14.0.dist-info}/RECORD

@@ -16,7 +16,7 @@ libinephany/observations/post_processors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 libinephany/observations/post_processors/postprocessors.py,sha256=43_e5UaDPr2KbAvqc_w3wLqnlm7bgRjqgCtyQ95-8cM,5913
 libinephany/pydantic_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/pydantic_models/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-libinephany/pydantic_models/configs/hyperparameter_configs.py,sha256=48pooGTxfg7Ol_mA770h1-x79Qgel-jxvlj5BRjcf1Q,13537
+libinephany/pydantic_models/configs/hyperparameter_configs.py,sha256=5BH7aPiFtSfhTKp-Z0WTbCYuGJvUTskgFgMttxpDZb0,13654
 libinephany/pydantic_models/configs/observer_config.py,sha256=v_ChzaVXC_rlZ7eDZPuCae1DdG7-PS3mPwC-OaWpGQo,1355
 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
@@ -24,10 +24,10 @@ libinephany/pydantic_models/schemas/agent_info.py,sha256=me5gDxvZjP9TNK588mpUvxi
 libinephany/pydantic_models/schemas/inner_task_profile.py,sha256=DBLdXeG_igZKp9t1A9bDiWPtaQ2r0rxKTGnpDpBjGlY,10344
 libinephany/pydantic_models/schemas/observation_models.py,sha256=YjQmrWZ0r-_LRp92jvhSD8p1grKsMVXCXoou4q15Ue8,1849
 libinephany/pydantic_models/schemas/request_schemas.py,sha256=VED8eAUvBofxeAx9gWU8DyCZOTVD3QsHRq-TO7kyOqk,1260
-libinephany/pydantic_models/schemas/response_schemas.py,sha256=7rEC9w8l-V-3hySUZUlHF-Pv0WPZaAwPCsgmlJsPalc,1524
+libinephany/pydantic_models/schemas/response_schemas.py,sha256=SKFuasdjX5aH_I0vT3SwnpwhyMf9cNPB1ZpDeAGgoO8,2158
 libinephany/pydantic_models/schemas/tensor_statistics.py,sha256=Z-x-Fi_Dm0pLoHI88DnJO1krY671o0zbGRzx-gXPtVY,7534
 libinephany/pydantic_models/states/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-libinephany/pydantic_models/states/hyperparameter_states.py,sha256=q2MUEb7jywa6SiGy0zF2if250AkKhUZeci2KfJJhQoA,31334
+libinephany/pydantic_models/states/hyperparameter_states.py,sha256=JWNSmQmncnQu2GTh0MeT8pvPLxttyrhsKXErucdURqQ,32223
 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
@@ -50,8 +50,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-0.13.8.dist-info/licenses/LICENSE,sha256=pogfDoMBP07ehIOvWymuWIar8pg2YLUhqOHsJQU3wdc,9250
-libinephany-0.13.8.dist-info/METADATA,sha256=cW9naaWcJk4FzvxOuhax_JgNqNInbOZKz-ffZQKRxuk,8354
-libinephany-0.13.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libinephany-0.13.8.dist-info/top_level.txt,sha256=bYAOXQdJgIoLkO2Ui0kxe7pSYegS_e38u0dMscd7COQ,12
-libinephany-0.13.8.dist-info/RECORD,,
+libinephany-0.14.0.dist-info/licenses/LICENSE,sha256=pogfDoMBP07ehIOvWymuWIar8pg2YLUhqOHsJQU3wdc,9250
+libinephany-0.14.0.dist-info/METADATA,sha256=32VAU-sCvWbhgj89cKZLRIWP_P9aa3QtH3jK9DuPnaE,8354
+libinephany-0.14.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libinephany-0.14.0.dist-info/top_level.txt,sha256=bYAOXQdJgIoLkO2Ui0kxe7pSYegS_e38u0dMscd7COQ,12
+libinephany-0.14.0.dist-info/RECORD,,