Skip to content

Commit 6af3211

Browse files
authored
Finally cleanup OMZ leftovers (#282)
1 parent 0d8cb49 commit 6af3211

File tree

3 files changed

+2
-18
lines changed

3 files changed

+2
-18
lines changed

.github/workflows/publish.yaml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,15 +24,13 @@ jobs:
2424
python -m pip install --upgrade build
2525
- name: Build sdist
2626
run: |
27-
sed -i '/omz_tools/d' src/python/pyproject.toml
2827
python -m build --sdist src/python/
2928
- uses: actions/upload-artifact@v4
3029
with:
3130
name: artifact-sdist
3231
path: src/python/dist/*.tar.gz
3332
- name: Build wheel
3433
run: |
35-
sed -i '/omz_tools/d' src/python/pyproject.toml
3634
python -m build --wheel src/python/
3735
- uses: actions/upload-artifact@v4
3836
with:

src/python/model_api/adapters/openvino_adapter.py

Lines changed: 2 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -171,21 +171,8 @@ def __init__(
171171
)
172172
self.model = core.read_model(self.model_path, weights_path)
173173
return
174-
if isinstance(model, str):
175-
from omz_tools.models import OMZModel, list_models
176-
177-
if model in list_models():
178-
omz_model = OMZModel.download(
179-
model,
180-
precision=precision,
181-
download_dir=download_dir,
182-
cache_dir=cache_dir,
183-
)
184-
self.model_path = omz_model.model_path
185-
log.info(f"Reading model {self.model_path}")
186-
self.model = core.read_model(self.model_path)
187-
return
188-
msg = "Model must be bytes, a file or existing OMZ model name"
174+
175+
msg = "Model must be bytes or a file"
189176
raise RuntimeError(msg)
190177

191178
def load_model(self) -> None:

src/python/pyproject.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@ dependencies = [
2828
"opencv-python-headless",
2929
"scipy>=1.5.4",
3030
"openvino>=2025.0",
31-
"omz_tools @ git+https://github.com/openvinotoolkit/open_model_zoo.git@master#egg=omz_tools&subdirectory=tools/model_tools",
3231
"pillow",
3332
]
3433

0 commit comments

Comments
 (0)