diff --git a/codecarbon/emissions_tracker.py b/codecarbon/emissions_tracker.py index 483c15285..b4d973c28 100644 --- a/codecarbon/emissions_tracker.py +++ b/codecarbon/emissions_tracker.py @@ -405,7 +405,7 @@ def _init_output_methods(self, api_key): if self._save_to_file: self._output_handlers.append( FileOutput( - os.path.join(self._output_dir, self._output_file), + self._output_file, self._output_dir, self._on_csv_write, ) diff --git a/codecarbon/output_methods/file.py b/codecarbon/output_methods/file.py index f5cce18a0..66b942386 100644 --- a/codecarbon/output_methods/file.py +++ b/codecarbon/output_methods/file.py @@ -15,15 +15,16 @@ class FileOutput(BaseOutput): Saves experiment artifacts to a file """ - def __init__(self, save_file_path: str, output_dir: str, on_csv_write: str = "append"): + def __init__(self, output_file_name: str, output_dir: str, on_csv_write: str = "append"): if on_csv_write not in {"append", "update"}: raise ValueError( f"Unknown `on_csv_write` value: {on_csv_write}" + " (should be one of 'append' or 'update'" ) + self.output_file_name: str = output_file_name self.output_dir: str = output_dir self.on_csv_write: str = on_csv_write - self.save_file_path: str = save_file_path + self.save_file_path = os.path.join(self.output_dir, self.output_file_name) logger.info( f"Saving emissions data to file {os.path.abspath(self.save_file_path)}" )