fmt

Author: Borda

.pre-commit-config.yaml

@@ -14,9 +14,9 @@ repos:
       - id: end-of-file-fixer
       - id: trailing-whitespace
       - id: check-case-conflict
+      - id: check-json
       - id: check-yaml
       - id: check-toml
-      - id: pretty-format-json
       - id: check-added-large-files
         args: ['--maxkb=250', '--enforce-all']
       - id: check-docstring-first
@@ -26,7 +26,7 @@ repos:
     rev: v2.31.1
     hooks:
       - id: pyupgrade
-        args: [--py36-plus]
+        args: [--py37-plus]
         name: Upgrade code
 
   - repo: https://github.com/myint/docformatter

templates/img-classify/classify.py

@@ -2,8 +2,9 @@
 import os
 
 import flash
+import matplotlib.pyplot as plt
 import pandas as pd
-import torch
+import seaborn as sn
 from flash.image import ImageClassificationData, ImageClassifier
 from pytorch_lightning.loggers import CSVLogger
 
@@ -36,10 +37,6 @@
 trainer.finetune(model, datamodule=datamodule, strategy="freeze")
 
 # %%
-
-import matplotlib.pyplot as plt
-import seaborn as sn
-
 metrics = pd.read_csv(f"{trainer.logger.log_dir}/metrics.csv")
 del metrics["step"]
 metrics.set_index("epoch", inplace=True)