humalab 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

humalab/utils.py

@@ -0,0 +1,37 @@
+import builtins
+
+# Define what counts as "standard" types
+STANDARD_TYPES = (
+    int, float, complex, bool, str, bytes, bytearray,
+    list, tuple, dict, set, frozenset, type(None)
+)
+
+def is_standard_type(obj, _seen=None):
+    """Recursively check if an object only contains standard Python types."""
+    if _seen is None:
+        _seen = set()
+
+    obj_id = id(obj)
+    if obj_id in _seen:
+        return True  # Prevent infinite recursion for self-referential structures
+    _seen.add(obj_id)
+
+    if isinstance(obj, STANDARD_TYPES):
+        if isinstance(obj, (list, tuple, set, frozenset)):
+            return all(is_standard_type(i, _seen) for i in obj)
+        elif isinstance(obj, dict):
+            return all(
+                is_standard_type(k, _seen) and is_standard_type(v, _seen)
+                for k, v in obj.items()
+            )
+        return True
+
+    # If it's a builtin constant type (e.g., Ellipsis, NotImplemented)
+    if obj in (Ellipsis, NotImplemented):
+        return True
+
+    # Check if it's a built-in type object (like int, str)
+    if isinstance(obj, type) and obj.__module__ == 'builtins':
+        return True
+
+    return False

{humalab-0.0.4.dist-info → humalab-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: humalab
-Version: 0.0.4
+Version: 0.0.6
 Summary: Python SDK for HumaLab - A platform for adaptive AI validation.
 Home-page: https://github.com/humalab/humalab_sdk
 Author: HumaLab Team

humalab-0.0.6.dist-info/RECORD

@@ -0,0 +1,39 @@
+humalab/__init__.py,sha256=d6RcDYAWf4HzzF_is-_nlTZXAg2UjLSoWRJegNOYz9U,416
+humalab/constants.py,sha256=gIdDoKnrAjfCMbnv4MnPncGYQXqELsaVYs3bQmH6k0s,852
+humalab/episode.py,sha256=WpWIxzkQSBm-YGCEnvjL2b5pEN5OQBzFZJb_b3f8mKM,6917
+humalab/humalab.py,sha256=FiC551LnCa1iQOvQ9tOUsbocs2Q2M3zy8gDgMAMFsJg,6712
+humalab/humalab_api_client.py,sha256=uy6Re-e4v-Z5lPhZnu_HeSOcxDfdxW1uaMOb5Dbu0JE,29549
+humalab/humalab_config.py,sha256=K-RC1MfAhuD_FCORySV7uyjKzW-bYf2tY6EOZslR0kg,2316
+humalab/humalab_test.py,sha256=YMwOIH-sAOMnxwq8yR-Ut4vYk5Ksix9zS9yUIxi3GU8,20725
+humalab/run.py,sha256=GSO_YrEkWqEBu7fvPjL1ru-xkEYMvXYXS2ReMm42V3k,10129
+humalab/utils.py,sha256=KMIPoduAbQzha9wvnyX8OjE8X-3TaNpecx2o-kgHbdw,1161
+humalab/assets/__init__.py,sha256=9WnZmRuS04q1GMAVfS8-SIjH5m50kmEnsKC2VAH1obA,167
+humalab/assets/archive.py,sha256=PALZRnpNExDfXiURVNEj6xxWqcW4nd9UBZ7QIHa8mI8,3119
+humalab/assets/resource_operator.py,sha256=YoKQ8_rfHDuvpn7zJsbaOIcouO5Oaehs0duFfB_rczg,4184
+humalab/assets/files/__init__.py,sha256=9XF-Zp2nldx060v8A3tMTcB5o5vJCUTFKF9rrIFZE_s,111
+humalab/assets/files/resource_file.py,sha256=HpPM4FVAya0kEggx8bFB__2S6CxyJyHyWZCQ6VcQLi4,1773
+humalab/assets/files/urdf_file.py,sha256=UehIGgF9qL5oFiMX81l3aolSca7rkzN_B6t6k3o51UE,2802
+humalab/dists/__init__.py,sha256=Q7zQbFGC_CnTgExMcnRtJdqJitqy1oBk4D6Qyvunlqc,387
+humalab/dists/bernoulli.py,sha256=4RSYBkeyTHkqOzkn-of4y_oNJPhhfkMOuW2APjNa8Eo,1975
+humalab/dists/categorical.py,sha256=plEIZIx5LRiaNreUCIk9EkhBD6eflxwMJy3XbseeUBw,2114
+humalab/dists/discrete.py,sha256=X2VufaTsI7x125ev7RL7Sc59JhWh51tlE2dXAD51XBU,2889
+humalab/dists/distribution.py,sha256=t0qTUjS_N0adiy_j2fdf-NHSlRs1pr0swpnszizs04I,1048
+humalab/dists/gaussian.py,sha256=eezw8qpcRIRpoSM7Vgnny8berqsbcfKm0v00Hxv9KI8,2592
+humalab/dists/log_uniform.py,sha256=mcJUYSoapxsIiynq_MZS_1NH1sxlC-Oupop2S3-EVHw,2626
+humalab/dists/truncated_gaussian.py,sha256=tB0BQHaiSDmB1eNb143dKHysbDEG3_pjuCzYBOy4ks8,4009
+humalab/dists/uniform.py,sha256=ouUAY8fvtu7azNCltU9g5t8jCw5OFQyYp8BlD7OWS5E,2545
+humalab/metrics/__init__.py,sha256=lI1QCl_8mws5yfxxrAHCoIj1n5WEtiTZB5mTGFqAPCo,200
+humalab/metrics/code.py,sha256=CybaYFX3KdJLFGshSY9Ss1jpkWwA5X2wD0hZNZJXUJw,713
+humalab/metrics/metric.py,sha256=3lw3386UE3-pWHsOcYXoEM_RBfFR38bqd8Ehzj7aByw,1970
+humalab/metrics/scenario_stats.py,sha256=lW-8tYrKh1nc_mclF7tg_o0UrhDoNgkVHPtabHgzdiQ,3004
+humalab/metrics/summary.py,sha256=Y6AbDJv7vF4N82JqkTmJuwyaJ9WnlgJ6aExT8SSqnGQ,2194
+humalab/scenarios/__init__.py,sha256=Q76ohPaMrU_NVwVVFQrxJIzIFjtibaOS60ljV1I54Os,148
+humalab/scenarios/scenario.py,sha256=SmknTAZ3f304DmOSAKIkaZljrZ-ucxHs062CQKtS7js,12326
+humalab/scenarios/scenario_operator.py,sha256=GxOB20UP-A6J5Y_ftHbWHpIrTMV6dx52r09Tr-WXfWU,3152
+humalab/scenarios/scenario_test.py,sha256=YYSzKfaeZVK0Kv09pVYgZvLP7Zrz02jgUOpQ-0btanc,24397
+humalab-0.0.6.dist-info/licenses/LICENSE,sha256=Gy0Nw_Z9pbrNSu-loW-PCDWJyrC8eWpIqqIGW-DFtp8,1064
+humalab-0.0.6.dist-info/METADATA,sha256=KqqLNrCKPoBkDNYRwUKxjZoS2jogk7pHLaKYBHcBjmE,1704
+humalab-0.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+humalab-0.0.6.dist-info/entry_points.txt,sha256=aY-hS7Kg8y5kGgYA14YmtTz5odBrgJIZ2fQMXAbVW_U,49
+humalab-0.0.6.dist-info/top_level.txt,sha256=hp7XXBDE40hi9T3Jx6mPFc6wJbAMzektD5VWXlSCW6o,8
+humalab-0.0.6.dist-info/RECORD,,

humalab/assets/resource_manager.py

@@ -1,57 +0,0 @@
-from humalab.assets.files.resource_file import ResourceFile
-from humalab.humalab_config import HumalabConfig
-from humalab.humalab_api_client import HumaLabApiClient
-from humalab.assets.files.urdf_file import URDFFile
-import os
-from typing import Any
-
-
-class ResourceManager:
-    def __init__(self,
-                 api_key: str | None = None,
-                 host: str | None = None,
-                 timeout: float | None = None):
-        self._humalab_config = HumalabConfig()
-        self._base_url = host or self._humalab_config.base_url
-        self._api_key = api_key or self._humalab_config.api_key
-        self._timeout = timeout or self._humalab_config.timeout
-
-        self._api_client = HumaLabApiClient(base_url=self._base_url,
-                                      api_key=self._api_key,
-                                      timeout=self._timeout)
-        
-    def _asset_dir(self, name: str, version: int) -> str:
-        return os.path.join(self._humalab_config.workspace_path, "assets", name, f"{version}")
-    
-    def _create_asset_dir(self, name: str, version: int) -> bool:
-        asset_dir = self._asset_dir(name, version)
-        if not os.path.exists(asset_dir):
-            os.makedirs(asset_dir, exist_ok=True)
-            return True
-        return False
-
-    def download(self,
-                 name: str, 
-                 version: int | None=None) -> Any:
-        resource = self._api_client.get_resource(name=name, version=version)
-        file_content = self._api_client.download_resource(name="lerobot")
-        filename = os.path.basename(resource['resource_url'])
-        filename = os.path.join(self._asset_dir(name, resource["version"]), filename)
-        if self._create_asset_dir(name, resource["version"]):
-            with open(filename, "wb") as f:
-                f.write(file_content)
-        
-        if resource["resource_type"].lower() == "urdf":
-            return URDFFile(name=resource["name"],
-                            version=resource["version"],
-                            description=resource.get("description"),
-                            filename=filename,
-                            urdf_filename=resource.get("filename"),
-                            created_at=resource.get("created_at"))
-
-        return ResourceFile(name=name, 
-                            version=resource["version"], 
-                            filename=filename,
-                            resource_type=resource["resource_type"],
-                            description=resource.get("description"),
-                            created_at=resource.get("created_at"))

humalab/metrics/dist_metric.py

@@ -1,22 +0,0 @@
-
-from humalab.metrics.metric import MetricGranularity, Metrics, MetricType
-
-class DistributionMetric(Metrics):
-    def __init__(self, 
-                 name: str, 
-                 distribution_type: str,
-                 episode_id: str,
-                 run_id: str,
-                 granularity: MetricGranularity = MetricGranularity.EPISODE) -> None:
-        """
-        Initialize the distribution metric.
-
-        Args:
-            name (str): The name of the metric.
-            distribution_type (str): The type of distribution (e.g., "normal", "uniform").
-            episode_id (str): The ID of the episode.
-            run_id (str): The ID of the run.
-            granularity (MetricGranularity): The granularity of the metric.
-        """
-        super().__init__(name, MetricType.DISTRIBUTION, episode_id=episode_id, run_id=run_id, granularity=granularity)
-        self.distribution_type = distribution_type

humalab/scenario.py

@@ -1,225 +0,0 @@
-from typing import Any
-import numpy as np
-from omegaconf import OmegaConf, ListConfig, AnyNode, DictConfig
-import yaml
-from humalab.dists.bernoulli import Bernoulli
-from humalab.dists.categorical import Categorical
-from humalab.dists.uniform import Uniform
-from humalab.dists.discrete import Discrete
-from humalab.dists.log_uniform import LogUniform
-from humalab.dists.gaussian import Gaussian
-from humalab.dists.truncated_gaussian import TruncatedGaussian
-from functools import partial
-from humalab.constants import EpisodeStatus
-from humalab.metrics.dist_metric import DistributionMetric
-from humalab.metrics.metric import MetricGranularity
-import copy
-import uuid
-
-DISTRIBUTION_MAP = {
-    "uniform": Uniform,
-    "bernoulli": Bernoulli,
-    "categorical": Categorical,
-    "discrete": Discrete,
-    "log_uniform": LogUniform,
-    "gaussian": Gaussian,
-    "truncated_gaussian": TruncatedGaussian,
-}
-
-DISTRIBUTION_PARAM_NUM_MAP = {
-    "uniform": 3,
-    "bernoulli": 2,
-    "categorical": 3,
-    "discrete": 4,
-    "log_uniform": 3,
-    "gaussian": 3,
-    "truncated_gaussian": 5,
-}
-
-class Scenario:
-    dist_cache = {}
-    def __init__(self) -> None:
-        self._generator = np.random.default_rng()
-        self._scenario_template = OmegaConf.create()
-        self._cur_scenario = OmegaConf.create()
-        self._scenario_id = None
-
-    def init(self,
-             run_id: str,
-             episode_id: str,
-             scenario: str | list | dict | None = None, 
-             seed: int | None=None, 
-             scenario_id: str | None=None,
-             num_env: int | None = None) -> None:
-        """
-        Initialize the scenario with the given parameters.
-        
-        Args:
-            run_id: The ID of the current run.
-            episode_id: The ID of the current episode.
-            scenario: The scenario configuration (YAML string, list, or dict).
-            seed: Optional seed for random number generation.
-            scenario_id: Optional scenario ID. If None, a new UUID is generated.
-            num_env: Optional number of parallel environments.
-        """
-        self._run_id = run_id
-        self._episode_id = episode_id
-        self._metrics = {}
-
-        self._num_env = num_env
-        self._scenario_id = scenario_id or str(uuid.uuid4())
-        self._generator = np.random.default_rng(seed)
-        self._configure()
-        scenario = scenario or {}
-        self._scenario_template = OmegaConf.create(scenario)
-        self.reset(episode_id=episode_id)
-
-    def _get_final_size(self, size: int | tuple[int, ...] | None) -> int | tuple[int, ...] | None:
-        n = self._num_env
-        if size is None:
-            return n
-        if n is None:
-            return size
-        if isinstance(size, int):
-            return (n, size)
-        return (n, *size)
-    
-    def _get_node_path(self, root: dict | list, node: str) -> str:
-        if isinstance(root, list):
-            root = {str(i): v for i, v in enumerate(root)}
-        
-        for key, value in root.items():
-            if value == node:
-                return str(key)
-            if isinstance(value, dict):
-                sub_path = self._get_node_path(value, node)
-                if sub_path:
-                    return f"{key}.{sub_path}"
-            elif isinstance(value, list):
-                for idx, item in enumerate(value):
-                    if item == node:
-                        return f"{key}[{idx}]"
-                    if isinstance(item, (dict, list)):
-                        sub_path = self._get_node_path(item, node)
-                        if sub_path:
-                            return f"{key}[{idx}].{sub_path}"
-        return ""
-
-    @staticmethod
-    def _convert_to_python(obj) -> Any:
-        if not isinstance(obj, (np.ndarray, np.generic)):
-            return obj
-
-        # NumPy scalar (np.generic) or 0-D ndarray
-        if isinstance(obj, np.generic) or (isinstance(obj, np.ndarray) and obj.ndim == 0):
-            return obj.item()
-
-        # Regular ndarray (1-D or higher)
-        if isinstance(obj, np.ndarray):
-            return obj.tolist()
-
-        return obj
-
-    def _configure(self) -> None:
-        self._clear_resolvers()
-        def distribution_resolver(dist_name: str, *args, _node_, _root_, _parent_, **kwargs):
-            if len(args) > DISTRIBUTION_PARAM_NUM_MAP[dist_name]:
-                args = args[:DISTRIBUTION_PARAM_NUM_MAP[dist_name]]
-                print(f"Warning: Too many parameters for {dist_name}, expected {DISTRIBUTION_PARAM_NUM_MAP[dist_name]}, got {len(args)}. Extra parameters will be ignored.")
-            
-            # print("_node_: ", _node_, type(_node_))
-            # print("_root_: ", _root_, type(_root_))
-            # print("_parent_: ", _parent_, type(_parent_))
-            # print("Args: ", args, len(args))
-            # print("Kwargs: ", kwargs)
-
-            root_yaml = yaml.safe_load(OmegaConf.to_yaml(_root_))
-            key_path = self._get_node_path(root_yaml, str(_node_))
-            # print("Key path: ", key_path)
-            
-            if key_path not in self._metrics:
-                self._metrics[key_path] = DistributionMetric(name=key_path,
-                                                            distribution_type=dist_name,
-                                                            run_id=self._run_id,
-                                                            episode_id=self._episode_id,
-                                                            granularity=MetricGranularity.EPISODE)
-
-            shape = None 
-            
-            if len(args) == DISTRIBUTION_PARAM_NUM_MAP[dist_name]:
-                shape = args[DISTRIBUTION_PARAM_NUM_MAP[dist_name] - 1]
-                args = args[:-1]
-            shape = self._get_final_size(shape)
-
-            key = str(_node_)
-            if key not in Scenario.dist_cache:
-                Scenario.dist_cache[key] = DISTRIBUTION_MAP[dist_name].create(self._generator, *args, size=shape, **kwargs)
-            ret_val = Scenario.dist_cache[key].sample()
-            ret_val = Scenario._convert_to_python(ret_val)
-
-            if isinstance(ret_val, list):
-                ret_val = ListConfig(ret_val)
-            self._metrics[key_path].log(ret_val)
-            return ret_val
-
-        for dist_name in DISTRIBUTION_MAP.keys():
-            OmegaConf.register_new_resolver(dist_name, partial(distribution_resolver, dist_name))
-
-    def _clear_resolvers(self) -> None:
-        self.dist_cache.clear()
-        OmegaConf.clear_resolvers()
-    
-    def __getattr__(self, name: Any) -> Any:
-        if name in self._cur_scenario:
-            return self._cur_scenario[name]
-        raise AttributeError(f"'Scenario' object has no attribute '{name}'")
-
-    def __getitem__(self, key: Any) -> Any:
-        if key in self._cur_scenario:
-            return self._cur_scenario[key]
-        raise KeyError(f"'Scenario' object has no key '{key}'")
-
-    def reset(self,
-              episode_id: str | None = None) -> None:
-        """Reset the scenario for a new episode.
-        
-        Args:
-            episode_id: Optional new episode ID. If None, keeps the current episode ID.
-        """
-        for metric in self._metrics.values():
-            metric.reset(episode_id=episode_id)
-        self._cur_scenario = copy.deepcopy(self._scenario_template)
-        OmegaConf.resolve(self._cur_scenario)
-    
-    def finish(self) -> None:
-        """Finish the scenario and submit final metrics.
-        """
-        for metric in self._metrics.values():
-            metric.finish()
-
-    @property
-    def template(self) -> Any:
-        """The template scenario configuration.
-        
-        Returns:
-            Any: The template scenario as an OmegaConf object.
-        """
-        return self._scenario_template
-    
-    @property
-    def cur_scenario(self) -> Any:
-        """The current scenario configuration.
-
-        Returns:
-            Any: The current scenario as an OmegaConf object.
-        """
-        return self._cur_scenario
-    
-    @property
-    def yaml(self) -> str:
-        """The current scenario configuration as a YAML string.
-
-        Returns:
-            str: The current scenario as a YAML string.
-        """
-        return OmegaConf.to_yaml(self._cur_scenario)

humalab-0.0.4.dist-info/RECORD

@@ -1,34 +0,0 @@
-humalab/__init__.py,sha256=LIL4eEPT_Boiond6pZegbAPvZwNX-ZWHHMVPkv3ehcU,140
-humalab/constants.py,sha256=YSiUI7R7g0gkImD6ezufRlOMdIBS5wJ--dLy11Qb2qY,135
-humalab/humalab.py,sha256=gudSi9nWLY09jJk-JJhAr0d5Gzmru4Y7FA2ttO7HXSQ,8617
-humalab/humalab_api_client.py,sha256=V1PSe1pVngfBUJgxWW-lJzrM1ElQH2ZIolHiDieXlec,8806
-humalab/humalab_config.py,sha256=CdjPgZFGYmmeYq5qsL84ZN8tGdoRP2TjaqS3wojKejg,2698
-humalab/humalab_test.py,sha256=X07HDp9JLllGMuJ2kYxs4WB1o1r1VErn30AAvmn4zuM,19005
-humalab/run.py,sha256=QloVqoZAWlBL5eYVCOqdIRWsjaUoY9BVQvd1BvNEgmY,7841
-humalab/scenario.py,sha256=Zc9ymN5VI-cSnXhX7KYQOM62FCCm8GeAYSyUiUBZl_8,8275
-humalab/scenario_test.py,sha256=9AyDj_l_43bO_ovhzhIvADTh4bAkq8YaMnPQ_sRPTog,28604
-humalab/assets/__init__.py,sha256=fTtwY08nfLxv7whDLGAMWRxwBJDrEZFb-1t19M4cxlc,138
-humalab/assets/archive.py,sha256=PALZRnpNExDfXiURVNEj6xxWqcW4nd9UBZ7QIHa8mI8,3119
-humalab/assets/resource_manager.py,sha256=GeVfjerdtvRL4njQO8a75bZaOf_h4KdFTwN8h103vsg,2605
-humalab/assets/files/__init__.py,sha256=9XF-Zp2nldx060v8A3tMTcB5o5vJCUTFKF9rrIFZE_s,111
-humalab/assets/files/resource_file.py,sha256=mvwmE13dQWUu4XGi4kL6hzsYadldt3fVkGubRZ1D-6w,1020
-humalab/assets/files/urdf_file.py,sha256=t91OdI5vvpv9-ELLyjrE9mkgd1TT2y3pPVpUYSXKZA0,2647
-humalab/dists/__init__.py,sha256=Q7zQbFGC_CnTgExMcnRtJdqJitqy1oBk4D6Qyvunlqc,387
-humalab/dists/bernoulli.py,sha256=6JdeuEPdoL2hVwcP94anmTbIDk_TsSeUctooCzugQag,1500
-humalab/dists/categorical.py,sha256=t0LwN8PKAXa2Qx4RjYRGbTs-6InT3s4Qbi0srr7iDZQ,2025
-humalab/dists/discrete.py,sha256=Yk-JebgGyBgmTrtibmaF5sJLIsdgsCSUwa8arO8XUHQ,2167
-humalab/dists/distribution.py,sha256=t0qTUjS_N0adiy_j2fdf-NHSlRs1pr0swpnszizs04I,1048
-humalab/dists/gaussian.py,sha256=ueGm8CLTj8cVxfU4fi5cQKZVnPRI6dBy_PLzUMqn2-A,1838
-humalab/dists/log_uniform.py,sha256=TeQdZA3JJHh5ptqEvKNLoOgTxafiSkCfF7HIThULo3Y,1868
-humalab/dists/truncated_gaussian.py,sha256=b7LP8O7l5fobdTutatkdGneA_FThurz7fnMpdX9csKs,2691
-humalab/dists/uniform.py,sha256=BBPlL10EUMAbV7UWuktfGlUDEZhRrqHYKi1qY92j7pk,1793
-humalab/metrics/__init__.py,sha256=e0PPkAMP5nW-kGfb67SjMMlgxK9Bkp7nQVD-JWoV-qw,246
-humalab/metrics/dist_metric.py,sha256=C7IpdFolw-VpkTHv-HTAK63kafH-WUjRTLxbF7h4W7g,921
-humalab/metrics/metric.py,sha256=W1mWQKEPVs9x257zzJvJQmAadRNnDqGAEU2BAq1skwM,4194
-humalab/metrics/summary.py,sha256=CloYeVkYgZAiaeM2gS6V8_tABukTkxAFJt0mpfmpbLI,2255
-humalab-0.0.4.dist-info/licenses/LICENSE,sha256=Gy0Nw_Z9pbrNSu-loW-PCDWJyrC8eWpIqqIGW-DFtp8,1064
-humalab-0.0.4.dist-info/METADATA,sha256=TlQItL7q4I-T3NcRMuLKCHcef5Tg85fgMOlY_AEtjbw,1704
-humalab-0.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-humalab-0.0.4.dist-info/entry_points.txt,sha256=aY-hS7Kg8y5kGgYA14YmtTz5odBrgJIZ2fQMXAbVW_U,49
-humalab-0.0.4.dist-info/top_level.txt,sha256=hp7XXBDE40hi9T3Jx6mPFc6wJbAMzektD5VWXlSCW6o,8
-humalab-0.0.4.dist-info/RECORD,,