truefoundry 0.4.4rc12__py3-none-any.whl → 0.4.5__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