(*args, **kwargs)
| 1543 | def _decorate(fn: Callable): |
| 1544 | @wraps(fn) |
| 1545 | def wrapped_fn(*args, **kwargs): |
| 1546 | fn_result = None |
| 1547 | |
| 1548 | # Handle backward compatibility for co2_signal_api_token |
| 1549 | _electricitymaps_token = electricitymaps_api_token |
| 1550 | if co2_signal_api_token is not _sentinel: |
| 1551 | logger.warning( |
| 1552 | "Parameter 'co2_signal_api_token' is deprecated and will be removed in a future version. " |
| 1553 | "Please use 'electricitymaps_api_token' instead." |
| 1554 | ) |
| 1555 | if electricitymaps_api_token is _sentinel: |
| 1556 | _electricitymaps_token = co2_signal_api_token |
| 1557 | |
| 1558 | if offline and offline is not _sentinel: |
| 1559 | if (country_iso_code is None or country_iso_code is _sentinel) and ( |
| 1560 | cloud_provider is None or cloud_provider is _sentinel |
| 1561 | ): |
| 1562 | raise Exception("Needs ISO Code of the Country for Offline mode") |
| 1563 | tracker = OfflineEmissionsTracker( |
| 1564 | project_name=project_name, |
| 1565 | measure_power_secs=measure_power_secs, |
| 1566 | output_dir=output_dir, |
| 1567 | output_file=output_file, |
| 1568 | output_methods=output_methods, |
| 1569 | save_to_file=save_to_file, |
| 1570 | save_to_logger=save_to_logger, |
| 1571 | logging_logger=logging_logger, |
| 1572 | save_to_prometheus=save_to_prometheus, |
| 1573 | save_to_logfire=save_to_logfire, |
| 1574 | prometheus_url=prometheus_url, |
| 1575 | output_handlers=output_handlers, |
| 1576 | gpu_ids=gpu_ids, |
| 1577 | electricitymaps_api_token=_electricitymaps_token, |
| 1578 | tracking_mode=tracking_mode, |
| 1579 | log_level=log_level, |
| 1580 | on_csv_write=on_csv_write, |
| 1581 | logger_preamble=logger_preamble, |
| 1582 | country_iso_code=country_iso_code, |
| 1583 | region=region, |
| 1584 | cloud_provider=cloud_provider, |
| 1585 | cloud_region=cloud_region, |
| 1586 | country_2letter_iso_code=country_2letter_iso_code, |
| 1587 | force_cpu_power=force_cpu_power, |
| 1588 | force_ram_power=force_ram_power, |
| 1589 | pue=pue, |
| 1590 | wue=wue, |
| 1591 | force_carbon_intensity_g_co2e_kwh=force_carbon_intensity_g_co2e_kwh, |
| 1592 | allow_multiple_runs=allow_multiple_runs, |
| 1593 | rapl_include_dram=rapl_include_dram, |
| 1594 | rapl_prefer_psys=rapl_prefer_psys, |
| 1595 | ) |
| 1596 | else: |
| 1597 | tracker = EmissionsTracker( |
| 1598 | project_name=project_name, |
| 1599 | measure_power_secs=measure_power_secs, |
| 1600 | api_call_interval=api_call_interval, |
| 1601 | api_endpoint=api_endpoint, |
| 1602 | api_key=api_key, |
nothing calls this directly
no test coverage detected
searching dependent graphs…