From 89ea404b64375a61ca8501e4441ac576e5a99b7e Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Tue, 23 Apr 2019 18:08:41 +0200
Subject: [PATCH] Fix tests

---
 bob/learn/tensorflow/test/test_db_to_tfrecords.py | 2 +-
 bob/learn/tensorflow/test/test_utils.py           | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/bob/learn/tensorflow/test/test_db_to_tfrecords.py b/bob/learn/tensorflow/test/test_db_to_tfrecords.py
index 2318ca59..db5491d0 100644
--- a/bob/learn/tensorflow/test/test_db_to_tfrecords.py
+++ b/bob/learn/tensorflow/test/test_db_to_tfrecords.py
@@ -107,7 +107,7 @@ def test_datasets_to_tfrecords():
     runner = CliRunner()
     with runner.isolated_filesystem():
         output_path = './test'
-        args = (dummy_config, '--outputs', output_path)
+        args = (dummy_config, '--output', output_path)
         result = runner.invoke(
             datasets_to_tfrecords, args=args, standalone_mode=False)
         assert_click_runner_result(result)
diff --git a/bob/learn/tensorflow/test/test_utils.py b/bob/learn/tensorflow/test/test_utils.py
index e4f61241..e08b5311 100644
--- a/bob/learn/tensorflow/test/test_utils.py
+++ b/bob/learn/tensorflow/test/test_utils.py
@@ -3,8 +3,8 @@
 # @author: Tiago de Freitas Pereira <tiago.pereira@idiap.ch>
 
 import numpy
-from bob.learn.tensorflow.utils import compute_embedding_accuracy, cdist,\
-     compute_embedding_accuracy_tensors, predict_using_tensors
+from bob.learn.tensorflow.utils import compute_embedding_accuracy, \
+     compute_embedding_accuracy_tensors
 
 import tensorflow as tf
 """
-- 
GitLab