From 89feea7d0060a650eb8a76752592ad991794319c Mon Sep 17 00:00:00 2001
From: Geremia Pompei <56674278+GeremiaPompei@users.noreply.github.com>
Date: Thu, 5 Dec 2024 19:26:42 +0100
Subject: [PATCH] fix: Remove dtype from uniform initialization inside of
 sparse initialization function

---
 torchdyno/models/initializers.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/torchdyno/models/initializers.py b/torchdyno/models/initializers.py
index 1f8e28d..4284325 100644
--- a/torchdyno/models/initializers.py
+++ b/torchdyno/models/initializers.py
@@ -222,9 +222,7 @@ def sparse(
     # use scipy.sparse.random to generate sparse random matrix
     npdtype = _torch_to_numpy_dtype(dtype)
     if values_sampler is None:
-        values_sampler = lambda x: np.random.uniform(
-            low=-1.0, high=1.0, size=x, dtype=npdtype
-        )
+        values_sampler = lambda x: np.random.uniform(low=-1.0, high=1.0, size=x)
     sparse_mat = scipy.sparse.random(
         shape[0],
         shape[1],