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