model_store to redis_model_store

Author: tylerhutcherson

docs/redis_model_store.ipynb

@@ -37,7 +37,7 @@
     "import os\n",
     "import redis\n",
     "\n",
-    "from model_store import ModelStore\n",
+    "from redis_model_store import ModelStore\n",
     "\n",
     "# Replace values below with your own if using Redis Cloud instance\n",
     "REDIS_HOST = os.getenv(\"REDIS_HOST\", \"localhost\") # ex: \"redis-18374.c253.us-central1-1.gce.cloud.redislabs.com\"\n",

pyproject.toml

@@ -14,7 +14,7 @@ classifiers = [
   "Programming Language :: Python :: 3.11",
   "License :: OSI Approved :: MIT License",
 ]
-packages = [{ include = "model_store", from = "." }]
+packages = [{ include = "redis_model_store", from = "." }]
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.13"

model_store/__init__.py renamed to redis_model_store/__init__.py

@@ -1,6 +1,6 @@
 from typing import Any, Iterator, List
 
-from model_store.serialize import PickleSerializer, SerializationError, Serializer
+from redis_model_store.serialize import PickleSerializer, SerializationError, Serializer
 
 
 class ModelShardManager:

model_store/serialize.py renamed to redis_model_store/serialize.py

@@ -7,8 +7,8 @@
 from redisvl.query import FilterQuery
 from redisvl.query.filter import FilterExpression, Tag
 
-from model_store.shard_manager import ModelShardManager
-from model_store.utils import (
+from redis_model_store.shard_manager import ModelShardManager
+from redis_model_store.utils import (
     PIPELINE_BATCH_SIZE,
     current_timestamp,
     new_model_version,

model_store/shard_manager.py renamed to redis_model_store/shard_manager.py

@@ -2,30 +2,30 @@
 
 
 def format():
-    subprocess.run(["isort", "./model_store", "./tests", "--profile", "black"], check=True)
-    subprocess.run(["black", "./model_store", "./tests"], check=True)
+    subprocess.run(["isort", "./redis_model_store", "./tests", "--profile", "black"], check=True)
+    subprocess.run(["black", "./redis_model_store", "./tests"], check=True)
 
 
 def check_format():
-    subprocess.run(["black", "--check", "./model_store"], check=True)
+    subprocess.run(["black", "--check", "./redis_model_store"], check=True)
 
 
 def sort_imports():
-    subprocess.run(["isort", "./model_store", "./tests/", "--profile", "black"], check=True)
+    subprocess.run(["isort", "./redis_model_store", "./tests/", "--profile", "black"], check=True)
 
 
 def check_sort_imports():
     subprocess.run(
-        ["isort", "./model_store", "--check-only", "--profile", "black"], check=True
+        ["isort", "./redis_model_store", "--check-only", "--profile", "black"], check=True
     )
 
 
 def check_lint():
-    subprocess.run(["pylint", "--rcfile=.pylintrc", "./model_store"], check=True)
+    subprocess.run(["pylint", "--rcfile=.pylintrc", "./redis_model_store"], check=True)
 
 
 def check_mypy():
-    subprocess.run(["python", "-m", "mypy", "./model_store"], check=True)
+    subprocess.run(["python", "-m", "mypy", "./redis_model_store"], check=True)
 
 
 def test():

model_store/store.py renamed to redis_model_store/store.py

@@ -1,6 +1,6 @@
 import pytest
 
-from model_store.serialize import PickleSerializer
+from redis_model_store.serialize import PickleSerializer
 
 
 @pytest.fixture

model_store/utils.py renamed to redis_model_store/utils.py

@@ -1,7 +1,7 @@
 import pytest
 
-from model_store.serialize import PickleSerializer
-from model_store.shard_manager import ModelShardManager, SerializationError
+from redis_model_store.serialize import PickleSerializer
+from redis_model_store.shard_manager import ModelShardManager, SerializationError
 
 
 @pytest.fixture