From f97b77192231a301188a02e7e35b104b8e4be596 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Wed, 14 Sep 2022 15:03:07 +0100
Subject: [PATCH] make --help display help

---
 aimodel/src/parse_args.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/src/parse_args.py b/aimodel/src/parse_args.py
index 3ed7a13..f5acb72 100644
--- a/aimodel/src/parse_args.py
+++ b/aimodel/src/parse_args.py
@@ -11,7 +11,7 @@ def parse_args():
 	"""Defines and parses the CLI arguments."""
 	
 	
-	if len(sys.argv) < 2:
+	if len(sys.argv) < 2 or sys.argv[1] == "--help":
 		sys.stderr.write("""
 This program trains, manipulates, visualises, and manages a contrastive learning based rainfall radar → water depth prediction model.
 It functions by first finding relationships between the rainfall radar data and the water depth + heightmap data (the 'pretrain' subcommand). After this, a decoder model to predict water depth (modelled as an image segmentation task), can then be trained.