Add type annotations so generate+autocli works

swagnercarena · Nov 18, 2022 · 9db46e1 · 9db46e1
1 parent 8f29e4c
commit 9db46e1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/paltas/generate.py b/paltas/generate.py
@@ -24,7 +24,11 @@
 from paltas.Utils.cli_maker import make_cli
 
 
-def generate_from_config(config_path, save_folder, n=1, save_png_too=False, tf_record=False):
+def generate_from_config(
+		config_path, save_folder, 
+		n: int = 1, 
+		save_png_too: bool = False, 
+		tf_record: bool = False):
 	"""Generate simulated strong lensing images
 
 	Args: