Merge branch 'master' into final

Conflicts: cellbox/cellbox/__init__.py cellbox/cellbox/dataset.py cellbox/cellbox/kernel.py cellbox/cellbox/model.py cellbox/cellbox/train.py cellbox/cellbox/utils.py
Mustardburger · Aug 27, 2023 · 8c8b83d · 8c8b83d
2 parents 991b52a + 0ae7b99
commit 8c8b83d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/cellbox/cellbox/__init__.py b/cellbox/cellbox/__init__.py
@@ -10,4 +10,3 @@
 from cellbox.version import __version__, VERSION, get_msg
 
 get_msg()
-#
diff --git a/cellbox/cellbox/version.py b/cellbox/cellbox/version.py
@@ -4,11 +4,14 @@
 
 __version__ = '0.1.0'
 VERSION = __version__
+# TODO(desmondyuan): update 0.3.3
+# * new function docstrings
+# * remove the binder implementation
+# * add basic tests
 
+def get_msg() -> None:
+    """Print the version history."""
 
-def get_msg():
-    """get version history"""
-    # for test: installation completed
     changelog = [
         """
         version 0.1.0

diff --git a/scripts/main_torch.py b/scripts/main_torch.py
@@ -20,12 +20,14 @@
 
 
 def set_seed(in_seed):
+    """Sets random seeds for numpy and tensorflow."""
     int_seed = int(in_seed)
     torch.manual_seed(int_seed)
     np.random.seed(int_seed)
 
 
 def prepare_workdir(in_cfg):
+    """Creates the working directory for each experiment and generates necessary files."""
     # Read Data
     in_cfg.root_dir = os.getcwd()
     in_cfg.node_index = pd.read_csv(in_cfg.node_index_file, header=None, names=None) \
Original file line number	Diff line number	Diff line change
Expand Up		@@ -10,4 +10,3 @@
		from cellbox.version import __version__, VERSION, get_msg

		get_msg()
		#