fix weight names to make

deNBI · Jun 10, 2020 · ef0a627 · ef0a627
1 parent 1f37985
commit ef0a627
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/project_usage_exporter.py b/project_usage_exporter.py
@@ -614,8 +614,8 @@ def main():
             try:
                 weight_response = requests.get(args.weight_update_endpoint)
                 current_weights = {
-                    x['resource_set_timestamp']: {'memory_mb': {y['value']: y['weight'] for y in x['ram_weights']},
-                                                  'vcpus': {y['value']: y['weight'] for y in x['vcpu_weights']}} for
+                    x['resource_set_timestamp']: {'memory_mb': {y['value']: y['weight'] for y in x['mb_usage']},
+                                                  'vcpus': {y['value']: y['weight'] for y in x['vcpus']}} for
                     x in weight_response.json()}
                 logging.info("Updated credits weights, new weights: " + str(current_weights))
                 exporter.update_weights(current_weights)