From b3345963f32da2cc60b448a949bc1aff061cac55 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Thu, 8 Dec 2022 18:58:32 +0000
Subject: [PATCH] missing arg pass

---
 aimodel/src/subcommands/train_mono.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aimodel/src/subcommands/train_mono.py b/aimodel/src/subcommands/train_mono.py
index e758237..8f86745 100644
--- a/aimodel/src/subcommands/train_mono.py
+++ b/aimodel/src/subcommands/train_mono.py
@@ -79,6 +79,7 @@ def run(args):
 		model_arch_enc	= args.arch_enc,
 		model_arch_dec	= args.arch_dec,
 		learning_rate	= args.learning_rate,
+		heightmap_input = args.heightmap is not None
 		
 		metadata = read_metadata(args.input),
 		# shape_water_out=[ args.water_size, args.water_size ], # The DESIRED output shape. the actual data will be cropped to match this.