From 970ac4d2da98cb26e40f1e38150638b81ad4c6e7 Mon Sep 17 00:00:00 2001 From: DylanAlloy Date: Fri, 19 Jul 2024 09:43:31 -0400 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A7=20fix(wip):=20hasty=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/conf.py | 2 +- magnet/base.py | 10 +++------- pyproject.toml | 2 +- setup.py | 2 +- 4 files changed, 6 insertions(+), 10 deletions(-) diff --git a/docs/conf.py b/docs/conf.py index a7ef85e..5a7509b 100644 --- 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 index bb12854..19ec362 100644 --- 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 index c1ae602..9e95529 100644 --- 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 index 28fed42..b6b0100 100644 --- 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',