Update linear_model.json

icfaust · web-flow · commit 595a7ee974b6 · 2024-10-07T23:22:22.000+02:00
diff --git a/configs/regular/bf16/linear_model.json b/configs/regular/bf16/linear_model.json
@@ -10,6 +10,9 @@
             "algorithm": {
                 "estimator": "LinearRegression",
                 "estimator_params": { "fit_intercept": true, "copy_X": true }
+            },
+            "data": {
+                "dtype": ["float32"]
             }
         },
         "sklearn linear parameters": {
