Update randomizeop.py

heptaflar · web-flow · commit 3a7007fa27a3 · 2025-05-06T20:35:40.000+02:00
diff --git a/kaleidoscope/operators/randomizeop.py b/kaleidoscope/operators/randomizeop.py
@@ -148,8 +148,8 @@ def run(self, source: Dataset) -> Dataset:  # noqa: D102
                 ],
                 dtype=z.dtype,
             )
-            target[v].attrs["kaleidoscope_entropy"] = np.array(
-                self.entropy(v, source_id)
+            target[v].attrs["entropy"] = np.array(
+                self.entropy(v, source_id), dtype=np.int64
             )
             get_logger().info(f"finished graph for variable: {v}")
         return target

Original file line number	Diff line number	Diff line change
`@@ -148,8 +148,8 @@ def run(self, source: Dataset) -> Dataset: # noqa: D102`
`148`	`148`	`],`
`149`	`149`	`dtype=z.dtype,`
`150`	`150`	`)`
`151`		`- target[v].attrs["kaleidoscope_entropy"] = np.array(`
`152`		`- self.entropy(v, source_id)`
	`151`	`+ target[v].attrs["entropy"] = np.array(`
	`152`	`+ self.entropy(v, source_id), dtype=np.int64`
`153`	`153`	`)`
`154`	`154`	`get_logger().info(f"finished graph for variable: {v}")`
`155`	`155`	`return target`