fix accidental rename

elazarg · Aug 4, 2024 · fbe1d96 · fbe1d96
1 parent 9a246ae
commit fbe1d96
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/checkpoint/persist.py b/checkpoint/persist.py
@@ -114,7 +114,7 @@ def commit(self, *args) -> None:
             pathlib.Path(temp_filename).rename(self.filename)
 
             with open(self.tsv_filename, "a") as f:
-                size = pickle.results((self.i, args, self.iterator)).__sizeof__()
+                size = pickle.dumps((self.i, args, self.iterator)).__sizeof__()
                 self.printing_index += 1
                 print(
                     self.printing_index,

diff --git a/scripts/vm.sh b/scripts/vm.sh
@@ -9,7 +9,10 @@ if [ -z "$EXPERIMENT" ]; then
   echo "Usage: $0 <experiment> [QMP_PORT] [TCP_PORT] [STEP]"
   exit 1
 fi
+
+mkdir -f results
 mkdir -f results/${EXPERIMENT}
+chmod -R a+w results
 
 STEP=${2:-1}
 QMP_PORT=${3:-4444}
@@ -91,6 +94,7 @@ write_files:
 runcmd:
   - sudo chown -R ubuntu:ubuntu ${GUEST_HOME}
   - [su, ubuntu, -c, "cp -r /mnt/${EXPERIMENT_TAG}/* ${GUEST_HOME}/"]
+  - [su, ubuntu, -c, "ln -s /mnt/results ./results"]
   - [su, ubuntu, -c, "python3 -m venv ${GUEST_HOME}/.venv"]
   - [su, ubuntu, -c, "${VENV_BIN}/pip install -r /mnt/${CHECKPOINT_LIB}/requirements.txt"]
   - [su, ubuntu, -c, "${VENV_BIN}/pip install -r ${GUEST_HOME}/${EXPERIMENT}/requirements.txt"]