🚧 fix(wip): hasty fix

Prismadic · Jul 19, 2024 · 970ac4d · 970ac4d
1 parent cce5218
commit 970ac4d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 10 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -11,7 +11,7 @@
 project = 'magnet'
 copyright = '2023, Prismadic, LLC'
 author = 'Prismadic, LLC.'
-release = '0.3.14'
+release = '0.3.15'
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

diff --git a/magnet/base.py b/magnet/base.py
@@ -1,17 +1,13 @@
 
-import nats, asyncio, docker, platform
+import nats, asyncio, docker, platform, os
 
-import platform
-
-# Check if the OS is macOS
-if platform.system() == 'Darwin':
-    from milvus import default_server
-
 from magnet.utils.globals import _f
 from magnet.utils.data_classes import MagnetConfig, IndexConfig
 
 milvus_server = default_server
 os_name = platform.system()
+if os_name == 'Darwin' and not os.getenv('DOCKER_ENV'):
+    from milvus import default_server
 
 auto_config = {
     "host": "127.0.0.1",

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "llm_magnet"
-version = "0.3.14"
+version = "0.3.15"
 description = "the small distributed language model toolkit. fine-tune state-of-the-art LLMs anywhere, rapidly."
 readme = "dynamic"
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='llm_magnet',
-    version='0.3.14',
+    version='0.3.15',
     description="the small distributed language model toolkit. fine-tune state-of-the-art LLMs anywhere, rapidly.",
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',