From 172337939d5713dd4872e6e756b56a9887e6ba3c Mon Sep 17 00:00:00 2001
From: Charlie Meyers <cmeyers@cs.umu.se>
Date: Wed, 29 Nov 2023 13:27:28 +0000
Subject: [PATCH] linting

---
 deckard/base/model/model.py | 7 ++++---
 deckard/layers/README.md    | 2 --
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/deckard/base/model/model.py b/deckard/base/model/model.py
index 22385897..d335ef60 100644
--- a/deckard/base/model/model.py
+++ b/deckard/base/model/model.py
@@ -203,7 +203,7 @@ def __call__(self, data: list, model: object, library=None):
                 raise e
         time_dict = {
             "train_time": (end - start) / 1e9,
-            "train_time_per_sample": (end - start)   / (len(data[0]) * 1e9),
+            "train_time_per_sample": (end - start) / (len(data[0]) * 1e9),
             "train_start_time": start_timestamp,
             "train_end_time": end_timestamp,
             "train_device": device,
@@ -644,7 +644,7 @@ def predict_proba(self, data=None, model=None, probabilities_file=None):
             predictions,
             {
                 "predict_proba_time": (end - start) / 1e9,
-                "predict_proba_time_per_sample": (end - start)   / (len(data[0]) * 1e9),
+                "predict_proba_time_per_sample": (end - start) / (len(data[0]) * 1e9),
                 "predict_proba_start_time": start_timestamp,
                 "predict_proba_stop_time": end_timestamp,
                 "predict_proba_device": device,
@@ -707,7 +707,8 @@ def predict_log_loss(self, data, model, losses_file=None):
             predictions,
             {
                 "predict_log_proba_time": (end - start) / 1e9,
-                "predict_log_proba_time_per_sample": (end - start)   / (len(data[0]) * 1e9),
+                "predict_log_proba_time_per_sample": (end - start)
+                / (len(data[0]) * 1e9),
                 "predict_log_proba_start_time": start_timestamp,
                 "predict_log_proba_stop_time": end_timestamp,
                 "predict_log_device": device,
diff --git a/deckard/layers/README.md b/deckard/layers/README.md
index 774c6707..2e8a75a1 100644
--- a/deckard/layers/README.md
+++ b/deckard/layers/README.md
@@ -14,5 +14,3 @@
 ├── afr.py: For building accelerated failure rate models
 ├── plots.py: For generating plots using seaborn
 └── watcher.py: Watches a file, transforms the data, and moves it somewhere else via parallelized scp (work in progress).
-
-