integrate-ai 9.12.0__tar.gz → 9.14.0__tar.gz

{integrate-ai-9.12.0 → integrate-ai-9.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: integrate-ai
-Version: 9.12.0
+Version: 9.14.0
 Summary: integrate.ai
 Author: integrate.ai
 License: Copyright (C) 2022 integrate.ai, Inc. All rights reserved.

{integrate-ai-9.12.0 → integrate-ai-9.14.0}/integrate_ai/client.py

@@ -11,7 +11,7 @@ from integrate_ai.utils.typer_utils import (
 from integrate_ai.utils.logger import get_log_paths, logs_folder
 from integrate_ai.utils.path_utils import get_volumes, get_mounted_path, get_aws_env
 
-batch_size_default = 16
+batch_size_default = 8192
 instruction_polling_time_default = 30
 log_interval_default = 10
 

{integrate-ai-9.12.0 → integrate-ai-9.14.0}/integrate_ai/utils/typer_utils.py

@@ -4,12 +4,12 @@ Module containing miscellaneous typer utilities.
 
 from functools import wraps
 from pathlib import Path
+from rich import print as rprint
 from typing import Callable, Any, Dict
 import typer
 import os
 import re
 import subprocess
-from integrate_ai.utils.error_handling import IntegrateAIException
 
 
 def toggle_prompt_wrapper(option_func: Callable, disable_prompts: str) -> Callable:
@@ -74,14 +74,15 @@ def validate_posix_path(path: Path) -> None:
 
 def show_package_version(package):
     show_version_command = f"pip show {package}"
-    print(f"Getting current version for {package}.")
+    rprint(f"Getting current version for {package}.")
 
     pipe = subprocess.Popen(show_version_command, stdout=subprocess.PIPE, shell=True)
     out, err = pipe.communicate()
     if pipe.returncode != 0:
-        raise IntegrateAIException("Failed to get version.", err)
+        rprint("Failed to get version.")
+        raise typer.Exit(code=1)
     out_string = out.decode("utf-8")
     for line in out_string.splitlines():
         if re.search("Version", line):
-            print(line)
+            rprint(line)
             break

{integrate-ai-9.12.0 → integrate-ai-9.14.0}/integrate_ai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: integrate-ai
-Version: 9.12.0
+Version: 9.14.0
 Summary: integrate.ai
 Author: integrate.ai
 License: Copyright (C) 2022 integrate.ai, Inc. All rights reserved.

{integrate-ai-9.12.0 → integrate-ai-9.14.0}/setup.py

@@ -70,5 +70,5 @@ iai client train --token <IAI_TOKEN> --session <SESSION_ID> --train_path <PATH_T
         'integrate_ai.utils',
     ),
     'python_requires': '>=3.7.5',
-    'version': '9.12.0',
+    'version': '9.14.0',
 })