truefoundry 0.4.4rc12__py3-none-any.whl → 0.5.0rc1__py3-none-any.whl

truefoundry/common/constants.py

@@ -31,6 +31,7 @@ class TrueFoundrySdkEnv(BaseSettings):
     TFY_INTERNAL_SIGNED_URL_SERVER_MAX_TIMEOUT: int = 5  # default: 5 seconds
     TFY_INTERNAL_SIGNED_URL_SERVER_DEFAULT_TTL: int = 3600  # default: 1 hour
     TFY_INTERNAL_SIGNED_URL_REQUEST_TIMEOUT: int = 3600  # default: 1 hour
+    TFY_INTERNAL_SIGNED_URL_CLIENT_LOG_LEVEL: str = "WARNING"
 
     # Artifacts
     TFY_ARTIFACTS_DOWNLOAD_CHUNK_SIZE_BYTES: int = 100 * 1000 * 1000
@@ -38,9 +39,13 @@ class TrueFoundrySdkEnv(BaseSettings):
     TFY_ARTIFACTS_UPLOAD_MAX_WORKERS: int = max(min(32, (os.cpu_count() or 2) * 2), 4)
     TFY_ARTIFACTS_DISABLE_MULTIPART_UPLOAD: bool = False
     TFY_ARTIFACTS_DOWNLOAD_FSYNC_CHUNKS: bool = False
-    TFY_CLI_LOCAL_DEV_MODE: bool = False
+
+    # Fo customizing the python image used for building via PythonBuild
     TFY_PYTHONBUILD_PYTHON_IMAGE_REPO: str = "public.ecr.aws/docker/library/python"
 
+    # For local development, this enables futher configuration via _TFYServersConfig
+    TFY_CLI_LOCAL_DEV_MODE: bool = False
+
 
 ENV_VARS = TrueFoundrySdkEnv()
 API_SERVER_RELATIVE_PATH = "api/svc"

truefoundry/common/utils.py

@@ -10,7 +10,6 @@ from truefoundry.common.constants import (
     MLFOUNDRY_SERVER_RELATIVE_PATH,
     TFY_HOST_ENV_KEY,
 )
-from truefoundry.logger import logger
 from truefoundry.pydantic_v1 import BaseSettings
 
 T = TypeVar("T")
@@ -103,20 +102,3 @@ def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
     tfy_host = tfy_host.strip("/")
     validate_tfy_host(tfy_host)
     return tfy_host
-
-
-def log_time(prefix: str = ""):
-    """Decorator to log the time taken by I/O operations."""
-
-    def decorator(func):
-        @wraps(func)
-        def wrapper(*args, **kwargs):
-            start_time = time.time()
-            result = func(*args, **kwargs)
-            elapsed_time = time.time() - start_time
-            logger.info(f"{prefix}{func.__name__} took {elapsed_time:.2f} seconds")
-            return result
-
-        return wrapper
-
-    return decorator

truefoundry/logger.py

@@ -15,3 +15,4 @@ def add_cli_handler(level: int = logging.INFO, show_path=False):
     handler.setLevel(level)
     logger.addHandler(handler)
     logger.setLevel(logging.DEBUG)
+    logger.propagate = False

truefoundry/ml/__init__.py

@@ -1,3 +1,4 @@
+from truefoundry.ml.autogen.client.models.library_name import LibraryName
 from truefoundry.ml.enums import (
     DataSlice,
     FileFormat,
@@ -10,15 +11,34 @@ from truefoundry.ml.log_types import Image, Plot
 from truefoundry.ml.log_types.artifacts.artifact import ArtifactPath, ArtifactVersion
 from truefoundry.ml.log_types.artifacts.dataset import DataDirectory, DataDirectoryPath
 from truefoundry.ml.log_types.artifacts.model import (
+    BlobStorageModelDirectory,
     ModelVersion,
 )
 from truefoundry.ml.logger import init_logger
 from truefoundry.ml.mlfoundry_api import get_client
 from truefoundry.ml.mlfoundry_run import MlFoundryRun
+from truefoundry.ml.model_framework import (
+    FastAIFramework,
+    GluonFramework,
+    H2OFramework,
+    KerasFramework,
+    LightGBMFramework,
+    ModelFrameworkType,
+    ONNXFramework,
+    PaddleFramework,
+    PyTorchFramework,
+    SklearnFramework,
+    SpaCyFramework,
+    StatsModelsFramework,
+    TensorFlowFramework,
+    TransformersFramework,
+    XGBoostFramework,
+)
 
 __all__ = [
     "ArtifactPath",
     "ArtifactVersion",
+    "BlobStorageModelDirectory",
     "DataDirectory",
     "DataDirectoryPath",
     "DataSlice",
@@ -32,6 +52,22 @@ __all__ = [
     "Plot",
     "ViewType",
     "get_client",
+    "FastAIFramework",
+    "GluonFramework",
+    "H2OFramework",
+    "KerasFramework",
+    "LightGBMFramework",
+    "ONNXFramework",
+    "PaddleFramework",
+    "PyTorchFramework",
+    "SklearnFramework",
+    "SpaCyFramework",
+    "StatsModelsFramework",
+    "TensorFlowFramework",
+    "TransformersFramework",
+    "XGBoostFramework",
+    "LibraryName",
+    "ModelFrameworkType",
 ]
 
 init_logger()