From 6430d5e4cee174c371581ad95b81219218ec961f Mon Sep 17 00:00:00 2001
From: Mark Keller <7525285+keller-mark@users.noreply.github.com>
Date: Fri, 3 Nov 2023 11:11:19 -0400
Subject: [PATCH] Use scratch

---
 demos/common.smk | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/demos/common.smk b/demos/common.smk
index 5929ad96..32213e97 100644
--- a/demos/common.smk
+++ b/demos/common.smk
@@ -1,8 +1,12 @@
 from pathlib import Path
+import platform
+
+# Check if this is running on O2
+IS_O2 = (platform.system() == "Linux")
 
 # Directory / file constants
-SRC_DIR = Path("src")
-DATA_DIR = Path("data")
+SRC_DIR = "src"
+DATA_DIR = ("data" if not IS_O2 else "/n/data1/hms/dbmi/gehlenborg/lab/cao-2020")
 RAW_DIR = DATA_DIR / "raw"
 PROCESSED_DIR = DATA_DIR / "processed"