libinephany 0.13.5__py3-none-any.whl → 0.13.7__py3-none-any.whl

libinephany/observations/observers/global_observers.py

@@ -36,6 +36,8 @@ class InitialHyperparameters(GlobalObserver):
 
         super().__init__(**kwargs)
 
+        force_skip = ["samples"]
+        skip_hparams = force_skip if skip_hparams is None else skip_hparams + force_skip
         self.skip_hparams = [] if skip_hparams is None else skip_hparams
         self.pad_with = pad_with
 

libinephany/observations/observers/local_observers.py

@@ -580,7 +580,7 @@ class ModuleTypeOneHot(LocalObserver):
         agent_modules = self.observer_config.agent_modules
         module_type = agent_modules[self.parameter_group_name]
 
-        if module_type in ModuleTypes:
+        if module_type in {field.value for field in ModuleTypes}:
             one_hot_index = ModuleTypes.get_index(module_type)
 
         else:

libinephany/pydantic_models/configs/outer_model_config.py

@@ -12,11 +12,7 @@
 
 try:
     # Re-export all classes from libipcode to maintain backward compatibility
-    from libipcode.pydantic_models.configs.outer_model_config import (
-        OuterModelConfig,
-        SharedActorCriticConfig,
-        SoleAgentConfig,
-    )
+    from libipcode.pydantic_models.configs.outer_model_config import OuterModelConfig, SharedActorCriticConfig
 except ImportError:
     # If libipcode is not available, provide a fallback or raise a more helpful error
     raise ImportError(
@@ -28,5 +24,4 @@ except ImportError:
 __all__ = [
     "OuterModelConfig",
     "SharedActorCriticConfig",
-    "SoleAgentConfig",
 ]

libinephany/utils/import_utils.py

@@ -14,6 +14,7 @@ from types import ModuleType
 # ======================================================================================================================
 
 WANDB = "wandb"
+LIGHTNING = "lightning"
 
 # ======================================================================================================================
 #
@@ -41,3 +42,11 @@ def try_import_wandb() -> ModuleType | None:
     """
 
     return try_to_import_module(module_name=WANDB)
+
+
+def try_import_lightning() -> ModuleType | None:
+    """
+    :return: None or the imported lightning module.
+    """
+
+    return try_to_import_module(module_name=LIGHTNING)

{libinephany-0.13.5.dist-info → libinephany-0.13.7.dist-info}/METADATA

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

{libinephany-0.13.5.dist-info → libinephany-0.13.7.dist-info}/RECORD

@@ -9,8 +9,8 @@ libinephany/observations/statistic_manager.py,sha256=LLg1zSxnJr2oQQepYla3qoUuRy1
 libinephany/observations/statistic_trackers.py,sha256=flkXquMHvY6YjnQAvRElsV5OUm7Ek_PhA1_fvtX-0oQ,30124
 libinephany/observations/observers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/observations/observers/base_observers.py,sha256=RkG5SW0b6Ooy0_oscRHxyB_YFNP7k8fxu37jBZElxIM,15418
-libinephany/observations/observers/global_observers.py,sha256=jkj9ZGi7LqDGkKASJ_2S-jQMwOFxBKyy0ePCKZDTH04,38537
-libinephany/observations/observers/local_observers.py,sha256=sZmM2Oa2VUGhsnN6LhcHSRB3AF8lzVdvAmmXnfWNP70,36976
+libinephany/observations/observers/global_observers.py,sha256=-BJJaYjQSO82qskIlY_iijd3Lk1Ei1d3Hg1fzmYUPSM,38659
+libinephany/observations/observers/local_observers.py,sha256=F6IId8STfQix7n9diap3YzqyAX_eR25KCcyxykkwJgM,37003
 libinephany/observations/observers/observer_containers.py,sha256=g73ScbRRVTNbGEBb-Nyk8AQwoDhKZaqTd6OYP8FIcOs,8771
 libinephany/observations/post_processors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 libinephany/observations/post_processors/postprocessors.py,sha256=43_e5UaDPr2KbAvqc_w3wLqnlm7bgRjqgCtyQ95-8cM,5913
@@ -18,7 +18,7 @@ libinephany/pydantic_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 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/observer_config.py,sha256=v_ChzaVXC_rlZ7eDZPuCae1DdG7-PS3mPwC-OaWpGQo,1355
-libinephany/pydantic_models/configs/outer_model_config.py,sha256=G1iYq0sLVExTTvwtMDcQoj9ZAY5l2TL9wYSWHIaY67Y,1282
+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=DBLdXeG_igZKp9t1A9bDiWPtaQ2r0rxKTGnpDpBjGlY,10344
@@ -38,7 +38,7 @@ libinephany/utils/dropout_utils.py,sha256=X43yCW7Dh1cC5sNnivgS5j1fn871K_RCvxCBTT
 libinephany/utils/enums.py,sha256=YH10mUhW4kjYS0cp4XUASok9vfPl0jv9ZhS3HpZD0Zg,2339
 libinephany/utils/error_severities.py,sha256=B9oidqOVaYOe0W6P6GwjpmuDsrkyTX30v1xdiUStCFk,1427
 libinephany/utils/exceptions.py,sha256=kgwLpHOgy3kciUz_I18xnYsWRtzdonfadUtwG2uDYk8,1823
-libinephany/utils/import_utils.py,sha256=m8wGHXBL-XxqH4NLFU3r4cjqmfQB-qtPWhBeGWXlvEA,1296
+libinephany/utils/import_utils.py,sha256=WzC6V6UIa0nCiU2MekROwG82fWBh9RuVzichtby5EvM,1495
 libinephany/utils/optim_utils.py,sha256=-PLqsyuq4ZH3spBy_olNB3yuLwvhnLrCF0384elCmXc,8777
 libinephany/utils/random_seeds.py,sha256=eF-ErrMShu8mp9V_gXrB_iUxR-Lb-OtHypEEUQAGn2Y,1565
 libinephany/utils/samplers.py,sha256=RxNib-BpFSsx0qRoWlkfRvUGJNhM-k2uhwXCYNYnThs,11546
@@ -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.5.dist-info/licenses/LICENSE,sha256=pogfDoMBP07ehIOvWymuWIar8pg2YLUhqOHsJQU3wdc,9250
-libinephany-0.13.5.dist-info/METADATA,sha256=yn_ZBP5TyIRTzfFH3onVhGHnNZLjYoYkck0Ts5i2Djo,8354
-libinephany-0.13.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-libinephany-0.13.5.dist-info/top_level.txt,sha256=bYAOXQdJgIoLkO2Ui0kxe7pSYegS_e38u0dMscd7COQ,12
-libinephany-0.13.5.dist-info/RECORD,,
+libinephany-0.13.7.dist-info/licenses/LICENSE,sha256=pogfDoMBP07ehIOvWymuWIar8pg2YLUhqOHsJQU3wdc,9250
+libinephany-0.13.7.dist-info/METADATA,sha256=JwaixywxnDQ5eFphN6QdK0-vryqsApvf_0YMHOv15zU,8354
+libinephany-0.13.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+libinephany-0.13.7.dist-info/top_level.txt,sha256=bYAOXQdJgIoLkO2Ui0kxe7pSYegS_e38u0dMscd7COQ,12
+libinephany-0.13.7.dist-info/RECORD,,