Fixed .pyx file

lyst · Feb 6, 2020 · e1ed5a7 · e1ed5a7
1 parent 6f47aae
commit e1ed5a7
Show file tree

Hide file tree

Showing 3 changed files with 8,139 additions and 7,711 deletions.
diff --git a/lightfm/_lightfm_fast.pyx.template b/lightfm/_lightfm_fast.pyx.template
@@ -1186,7 +1186,7 @@ def predict_lightfm(CSRMatrix item_features,
                     CSRMatrix user_features,
                     int[::1] user_ids,
                     int[::1] item_ids,
-                    double[::1] predictions,
+                    flt[::1] predictions,
                     FastLightFM lightfm,
                     int num_threads):
     """