diff --git a/lib/cli/RLTraderCLI.py b/lib/cli/RLTraderCLI.py index e20c0a1..b40e42b 100644 --- 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 index c2ccef1..40cac3d 100644 --- 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)