Remove state client

nada_ai/client/clients/__init__.py

@@ -1,5 +1,4 @@
-from .sklearn_client import SklearnClient
-from .state_client import StateClient
-from .torch_client import TorchClient
+from .sklearn import SklearnClient
+from .torch import TorchClient
 
-__all__ = ["SklearnClient", "StateClient", "TorchClient"]
+__all__ = ["SklearnClient", "TorchClient"]

tests/python-tests/test_model_client.py

@@ -7,7 +7,7 @@
 import nada_algebra as na
 from sklearn.linear_model import LinearRegression, LogisticRegression
 from torch import nn
-from nada_ai.client import ModelClient, StateClient, TorchClient, SklearnClient
+from nada_ai.client import ModelClient, TorchClient, SklearnClient
 import py_nillion_client as nillion
 
 
@@ -72,13 +72,6 @@ def test_sklearn_4(self):
         with pytest.raises(NotImplementedError):
             model_client.export_state_as_secrets("test_model", nillion.SecretInteger)
 
-    def test_state_client_1(self):
-        state_client = StateClient({"some_value": 1})
-
-        secrets = state_client.export_state_as_secrets("test_model", na.Rational)
-
-        assert list(secrets.keys()) == ["test_model_some_value_0"]
-
     def test_custom_client_1(self):
         class MyModelClient(ModelClient):
             def __init__(self) -> None: