From 564ed54344347439cef4c90d4062dd32b7e7ff36 Mon Sep 17 00:00:00 2001
From: Marius Ciepluch <11855163+norandom@users.noreply.github.com>
Date: Thu, 4 Apr 2024 15:47:41 +0000
Subject: [PATCH] prep script for goog colab env

---
 scripts/prepare_colab_env.sh | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 scripts/prepare_colab_env.sh

diff --git a/scripts/prepare_colab_env.sh b/scripts/prepare_colab_env.sh
new file mode 100644
index 0000000..ed8197f
--- /dev/null
+++ b/scripts/prepare_colab_env.sh
@@ -0,0 +1,16 @@
+#!/usr/bin/env bash
+
+if [ -d "/content" ]; then
+  echo "This appears to be a Google Colab environment."
+
+  echo "Installing dependencies"
+  pip install -r "https://raw.githubusercontent.com/norandom/project_bookworm/main/requirements.gpu.txt"
+
+  echo "Downloading data directly (workaround)"
+  cd "/content"
+  wget "https://github.com/norandom/project_bookworm/raw/main/data/export.documents.txt"
+  wget "https://github.com/norandom/project_bookworm/raw/main/data/export.txt"
+
+else
+  echo "This does not appear to be a Google Colab environment."
+fi