biolm-sdk 0.0.1__py2.py3-none-any.whl

biolm/__init__.py

@@ -0,0 +1,78 @@
+"""Top-level package for BioLM."""
+__author__ = """Nikhil Haas"""
+__email__ = "nikhil@biolm.ai"
+__version__ = '0.0.1'
+
+from biolm.core.http import BioLMApi, BioLMApiClient
+from biolm.client import BioLM
+from biolm.models import Model, predict, encode, generate
+from biolm.protocols import Protocol
+from biolm.workspaces import Workspace
+from biolm.volumes import Volume
+from biolm.examples import get_example, list_models
+from biolm.io import (
+    load_fasta,
+    to_fasta,
+    load_csv,
+    to_csv,
+    load_pdb,
+    to_pdb,
+    load_json,
+    to_json,
+)
+
+try:
+    from biolm import pipeline
+    _HAS_PIPELINE = True
+except ImportError:
+    _HAS_PIPELINE = False
+
+from typing import Optional, Union, List, Any
+
+__all__ = [
+    'BioLM',
+    'biolm',
+    'BioLMApi',
+    'BioLMApiClient',
+    'Model',
+    'Protocol',
+    'Workspace',
+    'Volume',
+    'predict',
+    'encode',
+    'generate',
+    'get_example',
+    'list_models',
+    'load_fasta',
+    'to_fasta',
+    'load_csv',
+    'to_csv',
+    'load_pdb',
+    'to_pdb',
+    'load_json',
+    'to_json',
+]
+if _HAS_PIPELINE:
+    __all__.append('pipeline')
+
+
+def biolm(
+    *,
+    entity: str,
+    action: str,
+    type: Optional[str] = None,
+    items: Union[Any, List[Any]],
+    params: Optional[dict] = None,
+    api_key: Optional[str] = None,
+    **kwargs
+) -> Any:
+    """Top-level convenience function that wraps the BioLM class and returns the result."""
+    return BioLM(
+        entity=entity,
+        action=action,
+        type=type,
+        items=items,
+        params=params,
+        api_key=api_key,
+        **kwargs
+    )