Fix/codecarbon (#242)

voorhs · github-actions[bot] · web-flow · commit 0e3c27880b3c · 2025-06-25T01:19:50.000+03:00
* fix name

* fix dependencies

* Update optimizer_config.schema.json

* change versions

---------

Co-authored-by: github-actions[bot] &lt;github-actions[bot]@users.noreply.github.com&gt;
diff --git a/autointent/_callbacks/emissions_tracker.py b/autointent/_callbacks/emissions_tracker.py
@@ -13,7 +13,7 @@
 class EmissionsTrackerCallback(OptimizerCallback):
     """Class for tracking energy consumption and carbon emissions."""
 
-    name = "emissions_tracker"
+    name = "codecarbon"
 
     current_module_name: str | None = None
 
diff --git a/docs/optimizer_config.schema.json b/docs/optimizer_config.schema.json
@@ -421,7 +421,7 @@
                                 "enum": [
                                     "wandb",
                                     "tensorboard",
-                                    "emissions_tracker"
+                                    "codecarbon"
                                 ],
                                 "type": "string"
                             },
diff --git a/pyproject.toml b/pyproject.toml
@@ -94,6 +94,7 @@ wandb = [
 ]
 codecarbon = [
     "codecarbon (>=3.0.2, <3.1.0)",
+    "pynvml (>=8.0.4, <12.0.0)", # to avoid "attribute nvmlDeviceGetTotalEnergyConsumption not found" error
 ]
 
 [project.urls]

Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,7 @@ wandb = [`
`94`	`94`	`]`
`95`	`95`	`codecarbon = [`
`96`	`96`	`"codecarbon (>=3.0.2, <3.1.0)",`
	`97`	`+ "pynvml (>=8.0.4, <12.0.0)", # to avoid "attribute nvmlDeviceGetTotalEnergyConsumption not found" error`
`97`	`98`	`]`
`98`	`99`
`99`	`100`	`[project.urls]`