From f79ab58e5d8db240086d59cd02b5859c83f2e41e Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Wed, 10 Jul 2019 20:47:02 +0800
Subject: [PATCH] update dependencies

---
 base/bootstrap.sh            | 5 +++++
 yao-scheduler/dev/Dockerfile | 1 +
 2 files changed, 6 insertions(+)

diff --git a/base/bootstrap.sh b/base/bootstrap.sh
index c97aefa..0742d82 100755
--- a/base/bootstrap.sh
+++ b/base/bootstrap.sh
@@ -12,6 +12,11 @@ if [ -d /workspace ]; then
   cd /workspace
 fi
 
+if ! grep -q *.py per_process_gpu_memory_fraction; then
+  echo "Resourse usage limitation not set!"
+  exit 1
+fi
+
 # use eval because commands likes `key=value command` would cause file not found error when using $@, but this eval will ruin current environment
 eval $@
 
diff --git a/yao-scheduler/dev/Dockerfile b/yao-scheduler/dev/Dockerfile
index 2afebd2..a313342 100644
--- a/yao-scheduler/dev/Dockerfile
+++ b/yao-scheduler/dev/Dockerfile
@@ -10,6 +10,7 @@ RUN wget https://dl.google.com/go/go1.12.4.linux-amd64.tar.gz && \
 ENV PATH $PATH:/usr/local/go/bin
 
 RUN go get github.com/Shopify/sarama
+RUN go get github.com/sirupsen/logrus
 
 RUN git clone https://github.com/newnius/YAO-scheduler.git /root/yao-scheduler/