Merge branch 'fix-dataset-set-weight-shoji-order'

Crunch-io · Jan 31, 2025 · d940c4f · d940c4f
2 parents be060c0 + 5ae1a13
commit d940c4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scrunch/datasets.py b/scrunch/datasets.py
@@ -2708,7 +2708,7 @@ def set_weight(self, variables=None):
             graph = [self[v].url for v in variables]
         if variables is None:
             graph = []
-        payload = {'graph': graph}
+        payload = {"element": "shoji:order", "graph": graph}
         return self.resource.variables.weights.patch(json.dumps(payload))
 
     @property
@@ -2740,7 +2740,7 @@ def remove_weight(self, variables):
                 raise NameError("%s was not found in weights" % variables)
 
         graph = [self[v].url for v in weights]
-        payload = {'graph': graph}
+        payload = {"element": "shoji:order", "graph": graph}
         return self.resource.variables.weights.patch(json.dumps(payload))
 
     def drop_rows(self, filter):