From 93ba8a91089e881f3286f9fd1cff0a794f2662d4 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Wed, 7 Oct 2020 17:46:06 +0200
Subject: [PATCH] Included train_utils.py in coverage report

---
 .gitlab-ci.yml        | 2 +-
 deepof/models.py      | 2 +-
 deepof/train_utils.py | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 1017e237..8a19ecd9 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -9,7 +9,7 @@ test:
     - pip install -r ./deepof/requirements.txt
     - pip install -e deepof/
     - coverage run --source deepof -m pytest
-    - coverage report -m --include deepof/utils.py,deepof/preprocess.py,deepof/model_utils.py,deepof/pose_utils.py,deepof/visuals.py,deepof/models.py,deepof/hypermodels.py
+    - coverage report -m --include deepof/utils.py,deepof/preprocess.py,deepof/model_utils.py,deepof/pose_utils.py,deepof/visuals.py,deepof/models.py,deepof/hypermodels.py,deepof/train_utils.py
     - coverage xml -o deepof_cov.xml
   artifacts:
     reports:
diff --git a/deepof/models.py b/deepof/models.py
index ad7fe4f5..f15f4a9c 100644
--- a/deepof/models.py
+++ b/deepof/models.py
@@ -315,7 +315,7 @@ class SEQ_2_SEQ_GMVAE:
                 ],
             )
 
-        else:
+        else:  # pragma: no cover
             raise NotImplementedError(
                 "Gaussian Mixtures are currently the only supported prior"
             )
diff --git a/deepof/train_utils.py b/deepof/train_utils.py
index ffa2e89a..8cd5fcb2 100644
--- a/deepof/train_utils.py
+++ b/deepof/train_utils.py
@@ -146,6 +146,7 @@ def tune_search(
 
     """
 
+    print(callbacks)
     tensorboard_callback, cp_callback, onecycle = callbacks
 
     if hypermodel == "S2SAE":
-- 
GitLab