Skip to content

Commit

Permalink
Fix ONNX issue on Mac OSX
Browse files Browse the repository at this point in the history
  • Loading branch information
al-rigazzi committed Oct 4, 2023
1 parent 546abf3 commit 8742857
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion tests/backends/test_cli_mini_exp.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,11 +27,23 @@

from contextlib import contextmanager

import pytest
import smartredis

import smartsim._core._cli.validate
from smartsim._core.utils.helpers import installed_redisai_backends

sklearn_available = True
try:
from skl2onnx import to_onnx
from sklearn.cluster import KMeans
from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestRegressor
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split

except ImportError:
sklearn_available = False

def test_cli_mini_exp_doesnt_error_out_with_dev_build(
local_db,
Expand Down Expand Up @@ -66,5 +78,5 @@ def _mock_make_managed_local_orc(*a, **kw):
# Test the backends the dev has installed
with_tf="tensorflow" in backends,
with_pt="torch" in backends,
with_onnx="onnxruntime" in backends,
with_onnx="onnxruntime" in backends and sklearn_available,
)

0 comments on commit 8742857

Please sign in to comment.