Fix tests.

notadamking · Jul 4, 2019 · cf84a3c · cf84a3c
1 parent f709e44
commit cf84a3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/cli/RLTraderCLI.py b/lib/cli/RLTraderCLI.py
@@ -11,7 +11,7 @@ def __init__(self):
         self.parser.add_argument("--input-data-path", "-t", type=str, default="data/input/coinbase-1h-btc-usd.csv")
         self.parser.add_argument("--pair", "-p", type=str, default="BTC/USD")
         self.parser.add_argument("--debug", "-n", action='store_false')
-        self.parser.add_argument('--mini-batches', type=int, default=1, help='Mini batches', dest='nminibatches')
+        self.parser.add_argument('--mini-batches', type=int, default=1, help='Mini batches', dest='n_minibatches')
         self.parser.add_argument('--train-split-percentage', type=int, default=0.8, help='Train set percentage')
         self.parser.add_argument('--verbose-model', type=int, default=1, help='Verbose model')
         self.parser.add_argument(

diff --git a/test/test_rl_trader.py b/test/test_rl_trader.py
@@ -20,7 +20,7 @@ def test_that_args_get_injected_correctly(self, data_mock, opt_mock, init_mock):
             assert(sut.tensorboard_path == args.tensorboard_path)
             assert(sut.params_db_path == args.params_db_path)
             assert(sut.model_verbose == args.model_verbose)
-            assert(sut.nminibatches == args.nminibatches)
+            assert(sut.n_minibatches == args.n_minibatches)
             assert(sut.train_split_percentage == args.train_split_percentage)
             assert(sut.input_data_path == args.input_data_path)
             assert(sut.model_verbose == args.model_verbose)