malevich-coretools 0.3.49__tar.gz → 0.3.51__tar.gz

{malevich_coretools-0.3.49/malevich_coretools.egg-info → malevich_coretools-0.3.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: malevich-coretools
-Version: 0.3.49
+Version: 0.3.51
 Author: Andrew Pogrebnoj
 Author-email: andrew@onjulius.co
 License-File: LICENSE

malevich_coretools-0.3.51/VERSION

@@ -0,0 +1 @@
+0.3.51

{malevich_coretools-0.3.49 → malevich_coretools-0.3.51}/malevich_coretools/funcs/funcs.py

@@ -890,6 +890,22 @@ async def get_userPipelines_realId_async(id: str, *args, **kwargs) -> Pipeline:
     return model_from_json(await send_to_core_get_async(USER_PIPELINES_REAL_ID(id), *args, **kwargs), Pipeline).simplify()
 
 
+def get_userPipelines_tagSearch(tag: str, *args, **kwargs) -> ResultIds:
+    return model_from_json(send_to_core_get(USER_PIPELINES_TAG(tag), *args, **kwargs), ResultIds)
+
+
+async def get_userPipelines_tagSearch_async(tag: str, *args, **kwargs) -> ResultIds:
+    return model_from_json(await send_to_core_get_async(USER_PIPELINES_TAG(tag), *args, **kwargs), ResultIds)
+
+
+def get_userPipelines_realIds_tagSearch(tag: str, *args, **kwargs) -> ResultIds:
+    return model_from_json(send_to_core_get(USER_PIPELINES_TAG_REAL_IDS(tag), *args, **kwargs), ResultIds)
+
+
+async def get_userPipelines_realIds_tagSearch_async(tag: str, *args, **kwargs) -> ResultIds:
+    return model_from_json(await send_to_core_get_async(USER_PIPELINES_TAG_REAL_IDS(tag), *args, **kwargs), ResultIds)
+
+
 def post_userPipelines(data: UserTask, wait: bool, *args, **kwargs) -> Alias.Id:
     return send_to_core_modify(USER_PIPELINES(wait), data, *args, **kwargs)
 

{malevich_coretools-0.3.49 → malevich_coretools-0.3.51}/malevich_coretools/funcs/helpers.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Coroutine, Dict, List, Literal, Optional, Union, overload
 from uuid import uuid4
 
 import pandas as pd
@@ -32,37 +32,89 @@ from malevich_coretools.secondary import Config, to_json
 __all__ = ["create_collection_from_file_df", "update_collection_from_file_df", "raw_collection_from_df", "raw_collection_from_file", "create_collection_from_df", "update_collection_from_df", "create_app_settings", "create_user_config", "create_task_component", "create_task_policy", "create_restrictions", "create_run_settings", "create_endpoint_override", "create_cfg_struct"]
 
 
+@overload
 def create_collection_from_file_df(
     file: str,
     name: Optional[str],
     metadata: Optional[Union[Dict[str, Any], str]],
     *args,
+    is_async: Literal[False],
     **kwargs
 ) -> Alias.Id:
+    pass
+
+
+@overload
+def create_collection_from_file_df(
+    file: str,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    *args,
+    is_async: Literal[True],
+    **kwargs
+) -> Coroutine[Any, Any, Alias.Id]:
+    pass
+
+
+def create_collection_from_file_df(
+    file: str,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    *args,
+    is_async: bool = False,
+    **kwargs
+) -> Union[Alias.Id, Coroutine[Any, Any, Alias.Id]]:
     try:
         data = pd.read_csv(file)
     except pd.errors.EmptyDataError:
         data = pd.DataFrame()
     if name is None:
         name = file
-    return create_collection_from_df(data, name=file, metadata=metadata, *args, **kwargs)
+    return create_collection_from_df(data, name=file, metadata=metadata, *args, is_async=is_async, **kwargs)
 
 
+@overload
 def update_collection_from_file_df(
     id: str,
     file: str,
     name: Optional[str],
     metadata: Optional[Union[Dict[str, Any], str]],
     *args,
+    is_async: Literal[False],
     **kwargs
 ) -> Alias.Id:
+    pass
+
+
+@overload
+def update_collection_from_file_df(
+    id: str,
+    file: str,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    *args,
+    is_async: Literal[True],
+    **kwargs
+) -> Coroutine[Any, Any, Alias.Id]:
+    pass
+
+
+def update_collection_from_file_df(
+    id: str,
+    file: str,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    *args,
+    is_async: bool = False,
+    **kwargs
+) -> Union[Alias.Id, Coroutine[Any, Any, Alias.Id]]:
     try:
         data = pd.read_csv(file)
     except pd.errors.EmptyDataError:
         data = pd.DataFrame()
     if name is None:
         name = file
-    return update_collection_from_df(id, data, name=file, metadata=metadata, *args, **kwargs)
+    return update_collection_from_df(id, data, name=file, metadata=metadata, *args, is_async=is_async, **kwargs)
 
 
 def raw_collection_from_df(
@@ -95,15 +147,41 @@ def raw_collection_from_file(
     return raw_collection_from_df(data, name, metadata)
 
 
+@overload
 def create_collection_from_df(
     data: pd.DataFrame,
     name: Optional[str],
     metadata: Optional[Union[Dict[str, Any], str]],
     batcher: Optional[Batcher] = None,
-    is_async: bool = False,
     *args,
+    is_async: Literal[False],
     **kwargs
 ) -> Alias.Id:
+    pass
+
+
+@overload
+def create_collection_from_df(
+    data: pd.DataFrame,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    batcher: Optional[Batcher] = None,
+    *args,
+    is_async: Literal[True],
+    **kwargs
+) -> Coroutine[Any, Any, Alias.Id]:
+    pass
+
+
+def create_collection_from_df(
+    data: pd.DataFrame,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    batcher: Optional[Batcher] = None,
+    *args,
+    is_async: bool = False,
+    **kwargs
+) -> Union[Alias.Id, Coroutine[Any, Any, Alias.Id]]:
     if batcher is None:
         batcher = Config.BATCHER
     data = raw_collection_from_df(data, name, metadata)
@@ -114,16 +192,44 @@ def create_collection_from_df(
     return post_collections_data(data, *args, **kwargs)
 
 
+@overload
 def update_collection_from_df(
     id: str,
     data: pd.DataFrame,
     name: Optional[str],
     metadata: Optional[Union[Dict[str, Any], str]],
     batcher: Optional[Batcher] = None,
-    is_async: bool = False,
     *args,
+    is_async: Literal[False],
     **kwargs
 ) -> Alias.Id:
+    pass
+
+
+@overload
+def update_collection_from_df(
+    id: str,
+    data: pd.DataFrame,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    batcher: Optional[Batcher] = None,
+    *args,
+    is_async: Literal[True],
+    **kwargs
+) -> Coroutine[Any, Any, Alias.Id]:
+    pass
+
+
+def update_collection_from_df(
+    id: str,
+    data: pd.DataFrame,
+    name: Optional[str],
+    metadata: Optional[Union[Dict[str, Any], str]],
+    batcher: Optional[Batcher] = None,
+    *args,
+    is_async: bool = False,
+    **kwargs
+) -> Union[Alias.Id, Coroutine[Any, Any, Alias.Id]]:
     if batcher is None:
         batcher = Config.BATCHER
     data = raw_collection_from_df(data, name, metadata)

{malevich_coretools-0.3.49 → malevich_coretools-0.3.51}/malevich_coretools/secondary/const.py

@@ -139,6 +139,8 @@ USER_PIPELINES_MAP_IDS = f"{USER_PIPELINES_MAIN}/mapIds"
 USER_PIPELINES_MAP_ID = lambda id: f"{USER_PIPELINES_MAIN}/mapIds/{urllib.parse.quote(str(id), safe='')}"
 USER_PIPELINES_ID = lambda id, wait: with_wait(f"{USER_PIPELINES_MAIN}/{urllib.parse.quote(str(id), safe='')}", wait)
 USER_PIPELINES_REAL_ID = lambda id: f"{USER_PIPELINES_MAIN}/realIds/{urllib.parse.quote(str(id), safe='')}"
+USER_PIPELINES_TAG = lambda tag: f"{USER_PIPELINES_MAIN}/tagSearch/{urllib.parse.quote(str(tag), safe='')}"
+USER_PIPELINES_TAG_REAL_IDS = lambda tag: f"{USER_PIPELINES_MAIN}/realIds/tagSearch/{urllib.parse.quote(str(tag), safe='')}"
 
 ## UserCfgsController
 USER_CFGS_MAIN = f"{API_VERSION}/userCfgs"