chore: fix pcc

zama-ai · Oct 20, 2023 · be56a3a · be56a3a
1 parent 9acc08f
commit be56a3a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/benchmarks/common.py b/benchmarks/common.py
@@ -342,6 +342,7 @@ def should_test_config_in_fhe(
         "TweedieRegressor",
         "PoissonRegressor",
         "GammaRegressor",
+        "SGDRegressor",
     }:
         return True
 

diff --git a/tests/sklearn/test_dump_onnx.py b/tests/sklearn/test_dump_onnx.py
@@ -397,6 +397,15 @@ def test_dump(
 ) {
   %variable = Gemm[alpha = 1, beta = 1](%input_0, %_operators.0.coefficients, %_operators.0.intercepts)
   return %variable
+}""",
+        "SGDRegressor": """graph torch_jit (
+  %input_0[DOUBLE, symx10]
+) initializers (
+  %_operators.0.coefficients[FLOAT, 10x1]
+  %_operators.0.intercepts[FLOAT, 1]
+) {
+  %variable = Gemm[alpha = 1, beta = 1](%input_0, %_operators.0.coefficients, %_operators.0.intercepts)
+  return %variable
 }""",
         "Lasso": """graph torch_jit (
   %input_0[DOUBLE, symx10]