updated to Ray 2.7

2023-09-18 22:03:20 -07:00
parent 71b3d50a05
commit b98bd5b1ae
15 changed files with 3484 additions and 2086 deletions
--- a/tests/model/conftest.py
+++ b/tests/model/conftest.py
@@ -1,7 +1,7 @@
 import pytest
-from ray.train.torch.torch_predictor import TorchPredictor

 from madewithml import predict
+from madewithml.predict import TorchPredictor


 def pytest_addoption(parser):