[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Lightning-AI · Sep 15, 2023 · a2fb364 · a2fb364
1 parent 2807139
commit a2fb364
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/tests/tests_fabric/loggers/test_csv.py b/tests/tests_fabric/loggers/test_csv.py
@@ -16,7 +16,6 @@
 
 import pytest
 import torch
-import csv
 
 from lightning.fabric.loggers import CSVLogger
 from lightning.fabric.loggers.csv_logs import _ExperimentWriter
@@ -131,13 +130,13 @@ def test_append_columns(tmp_path):
     # new key appears
     metrics = {"a": 1, "b": 2, "c": 3}
     logger.log_metrics(metrics)
-    with open(logger.experiment.metrics_file_path, "r") as file:
+    with open(logger.experiment.metrics_file_path) as file:
         header = file.readline().strip()
         assert set(header.split(",")) == {"step", "a", "b", "c"}
 
     # key disappears
     metrics = {"a": 1, "c": 3}
     logger.log_metrics(metrics)
-    with open(logger.experiment.metrics_file_path, "r") as file:
+    with open(logger.experiment.metrics_file_path) as file:
         header = file.readline().strip()
         assert set(header.split(",")) == {"step", "a", "b", "c"}
diff --git a/tests/tests_pytorch/loggers/test_csv.py b/tests/tests_pytorch/loggers/test_csv.py
@@ -149,13 +149,13 @@ def test_append_columns(tmp_path):
     # new key appears
     metrics = {"a": 1, "b": 2, "c": 3}
     logger.log_metrics(metrics)
-    with open(logger.experiment.metrics_file_path, "r") as file:
+    with open(logger.experiment.metrics_file_path) as file:
         header = file.readline().strip()
         assert set(header.split(",")) == {"step", "a", "b", "c"}
 
     # key disappears
     metrics = {"a": 1, "c": 3}
     logger.log_metrics(metrics)
-    with open(logger.experiment.metrics_file_path, "r") as file:
+    with open(logger.experiment.metrics_file_path) as file:
         header = file.readline().strip()
         assert set(header.split(",")) == {"step", "a", "b", "c"}