From e7b24475b58245418bfcbe9b90ee0cdf9ffc7a6c Mon Sep 17 00:00:00 2001
From: David Hendriks <davidhendriks93@gmail.com>
Date: Fri, 11 Nov 2022 13:34:11 +0000
Subject: [PATCH] working towards putting all the montecarlo things

---
 binarycpython/utils/population_extensions/dataIO.py          | 2 +-
 .../utils/population_extensions/sampling_variables.py        | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/binarycpython/utils/population_extensions/dataIO.py b/binarycpython/utils/population_extensions/dataIO.py
index ce459647b..ae943a0c4 100644
--- a/binarycpython/utils/population_extensions/dataIO.py
+++ b/binarycpython/utils/population_extensions/dataIO.py
@@ -410,7 +410,7 @@ class dataIO:
 
         Also, make sure that in this export there are the basic parameters
         like m1,m2,sep, orb-per, ecc, probability etc.
-
+        TODO: this function can probably be cleaned a bit and can rely on the other startup and clean up functions (see population_class)
         On default this will write to the datadir, if it exists
 
         Args:
diff --git a/binarycpython/utils/population_extensions/sampling_variables.py b/binarycpython/utils/population_extensions/sampling_variables.py
index f8c995d81..6e4cc8e04 100644
--- a/binarycpython/utils/population_extensions/sampling_variables.py
+++ b/binarycpython/utils/population_extensions/sampling_variables.py
@@ -147,6 +147,7 @@ class sampling_variables:
         bottomcode: Union[str, None] = None,
         condition: Union[str, None] = None,
         dry_parallel: Union[bool, None] = False,
+        dependency_variables: Union[list, None] = None,
     ) -> None:
         """
         Function to add sampling variables to the grid_options.
@@ -248,6 +249,9 @@ class sampling_variables:
 
             bottomcode:
                 Code added at the very bottom of the block.
+
+            dependency_variables:
+                TODO: describe
         """
 
         # check parameters
@@ -278,6 +282,7 @@ class sampling_variables:
             "bottomcode": bottomcode,
             "sampling_variable_number": len(self.grid_options["_sampling_variables"]),
             "dry_parallel": dry_parallel,
+            "dependency_variables": dependency_variables,
         }
 
         # Check for gridtype input
-- 
GitLab