postgresml
diff --git a/‎pgml-extension/Cargo.lock
Lines changed: 1 addition & 1 deletion b/‎pgml-extension/Cargo.lock
Lines changed: 1 addition & 1 deletion
diff --git a/‎pgml-extension/Cargo.toml
Lines changed: 1 addition & 1 deletion b/‎pgml-extension/Cargo.toml
Lines changed: 1 addition & 1 deletion
diff --git a/‎pgml-extension/sql/pgml--2.4.2--2.4.3.sql b/‎pgml-extension/sql/pgml--2.4.2--2.4.3.sql
diff --git a/‎pgml-extension/src/bindings/transformers.py
Lines changed: 2 additions & 1 deletion b/‎pgml-extension/src/bindings/transformers.py
Lines changed: 2 additions & 1 deletion
@@ -1,6 +1,6 @@
 [package]
 name = "pgml"
-version = "2.4.2"
+version = "2.4.3"
 edition = "2021"
 
 [lib]
 
@@ -53,9 +53,10 @@ def default(self, obj):
 def transform(task, args, inputs, cache):
     task = json.loads(task)
     args = json.loads(args)
-    args["device"] = assign_device(args.get("device"))
     inputs = json.loads(inputs)
 
+    task["device"] = assign_device(task.get("device"))
+
     if cache:
         key = ",".join([f"{key}:{val}" for (key, val) in sorted(task.items())])
         if key not in __cache_transform_pipeline_by_task: