diff --git a/src/neptune_scale/__init__.py b/src/neptune_scale/__init__.py index 2bc91f16..15e64e09 100644 --- a/src/neptune_scale/__init__.py +++ b/src/neptune_scale/__init__.py @@ -442,7 +442,7 @@ def log_metrics( """ self.log(step=step, timestamp=timestamp, metrics=data) - def log_configs(self, data: Optional[Dict[str, Union[float, bool, int, str, datetime, list, set]]] = None) -> None: + def log_configs(self, data: Optional[Dict[str, Union[float, bool, int, str, datetime]]] = None) -> None: """ Logs the specified metadata to a Neptune run. @@ -516,7 +516,7 @@ def log( self, step: Optional[Union[float, int]] = None, timestamp: Optional[datetime] = None, - configs: Optional[Dict[str, Union[float, bool, int, str, datetime, list, set]]] = None, + configs: Optional[Dict[str, Union[float, bool, int, str, datetime]]] = None, metrics: Optional[Dict[str, Union[float, int]]] = None, tags_add: Optional[Dict[str, Union[List[str], Set[str]]]] = None, tags_remove: Optional[Dict[str, Union[List[str], Set[str]]]] = None, @@ -548,7 +548,7 @@ def log( verify_collection_type("`tags_add` keys", list(tags_add.keys()), str) verify_collection_type("`tags_remove` keys", list(tags_remove.keys()), str) - verify_collection_type("`configs` values", list(configs.values()), (float, bool, int, str, datetime, list, set)) + verify_collection_type("`configs` values", list(configs.values()), (float, bool, int, str, datetime)) verify_collection_type("`metrics` values", list(metrics.values()), (float, int)) verify_collection_type("`tags_add` values", list(tags_add.values()), (list, set)) verify_collection_type("`tags_remove` values", list(tags_remove.values()), (list, set)) diff --git a/src/neptune_scale/core/metadata_splitter.py b/src/neptune_scale/core/metadata_splitter.py index 2aec1ea7..c2018926 100644 --- a/src/neptune_scale/core/metadata_splitter.py +++ b/src/neptune_scale/core/metadata_splitter.py @@ -43,7 +43,7 @@ def __init__( run_id: str, step: Optional[Union[int, float]], timestamp: datetime, - fields: Dict[str, Union[float, bool, int, str, datetime, list, set]], + fields: Dict[str, Union[float, bool, int, str, datetime]], metrics: Dict[str, float], add_tags: Dict[str, Union[List[str], Set[str]]], remove_tags: Dict[str, Union[List[str], Set[str]]],