From b1c152179acea6813fc701dbfbb9e9385504166c Mon Sep 17 00:00:00 2001
From: Philipp Horstenkamp <phorstenkamp@fuzzy.de>
Date: Tue, 12 Jul 2022 20:04:34 +0200
Subject: [PATCH] Cleand up the data collection

---
 data/.gitignore                    | 2 ++
 data/collected_and_filtered.pickle | 3 ---
 data/collected_routes_np.pickle    | 3 ---
 docker-compose.yaml                | 1 +
 pyrate/pyrate.dockerfile           | 2 +-
 5 files changed, 4 insertions(+), 7 deletions(-)
 delete mode 100644 data/collected_and_filtered.pickle
 delete mode 100644 data/collected_routes_np.pickle

diff --git a/data/.gitignore b/data/.gitignore
index 5b015e4..cdde138 100644
--- a/data/.gitignore
+++ b/data/.gitignore
@@ -2,3 +2,5 @@ tmp_*.pickle
 collected.pickle
 test.pickle
 collectedWithImage.pickle
+collected_and_filtered.pickle
+collected_routes_np.pickle
diff --git a/data/collected_and_filtered.pickle b/data/collected_and_filtered.pickle
deleted file mode 100644
index d02811b..0000000
--- a/data/collected_and_filtered.pickle
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:37605e699ef684e5dc4ba0d0e63a55337fb3158e2b65538a7b108b29648f8b4e
-size 9895843
diff --git a/data/collected_routes_np.pickle b/data/collected_routes_np.pickle
deleted file mode 100644
index bf617e0..0000000
--- a/data/collected_routes_np.pickle
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:45214e2e1addb9869357bb175ef0f609d372c6ecb4adfd4c7610a6add2b4c55f
-size 112607399
diff --git a/docker-compose.yaml b/docker-compose.yaml
index 0a90c58..72167db 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -8,6 +8,7 @@ services:
     container_name: pyrate
     ports:
     - "8888:8888"
+    - "6006:6006"
     volumes:
     - ./experiments:/pyrate/experiments
     - ./data:/pyrate/data
diff --git a/pyrate/pyrate.dockerfile b/pyrate/pyrate.dockerfile
index a5bc2fb..8be8781 100644
--- a/pyrate/pyrate.dockerfile
+++ b/pyrate/pyrate.dockerfile
@@ -9,7 +9,7 @@ RUN apt install graphviz -yqq
 
 # Install additinal requirements
 RUN pip3 install jupyter tensorflow tensorboard keras-tuner black[jupyter] mapply humanize jupyter-resource-usage
-RUN pip3 install tensorflow-addons
+RUN pip3 install tensorflow-addons pydot pydotplus
 RUN pip install jupyter_contrib_nbextensions
 RUN jupyter contrib nbextension install --sys-prefix
 RUN jupyter nbextension enable scratchpad/main --sys-prefix