From 5b11fa56561b73bbd29d43da1721f42d991bf1f5 Mon Sep 17 00:00:00 2001
From: Zach Dwiel <zach.dwiel@intel.com>
Date: Sun, 18 Nov 2018 03:57:30 -0500
Subject: [PATCH] check for local mujoco key in build process (#105)

approved by scott.
---
 docker/Makefile | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/docker/Makefile b/docker/Makefile
index d101b7a..757a307 100644
--- a/docker/Makefile
+++ b/docker/Makefile
@@ -2,6 +2,7 @@
 REGISTRY=docker.io
 ORGANIZATION=nervana
 IMAGE=coach
+CONTEXT = $(realpath ..)
 
 BUILD_ARGUMENTS=
 RUN_ARGUMENTS=
@@ -14,6 +15,14 @@ ifdef https_proxy
 	BUILD_ARGUMENTS+=--build-arg https_proxy=$(https_proxy)
 	RUN_ARGUMENTS+=--env https_proxy=$(https_proxy)
 endif
+
+ifndef MUJOCO_KEY
+KEY_NAME=${CONTEXT}/mjkey.txt
+ifneq ("$(wildcard ${KEY_NAME})", "")
+MUJOCO_KEY:=$(shell cat ${KEY_NAME} | base64 -w 0)
+endif
+endif
+
 ifdef MUJOCO_KEY
 	BUILD_ARGUMENTS+=--build-arg MUJOCO_KEY=$(MUJOCO_KEY)
 endif
@@ -27,13 +36,11 @@ INTEGRATION_TESTS=python3 -m pytest rl_coach/tests -m integration_test --tb=shor
 ifdef PRESETS
 	PRESETS := -p $(PRESETS)
 else
-	PRESETS := 
+	PRESETS :=
 endif
 GOLDEN_TESTS=python3 rl_coach/tests/test_golden.py ${PRESETS}
 TRACE_TESTS=python3 rl_coach/tests/trace_tests.py -prl ${PRESETS}
 
-CONTEXT = $(realpath ..)
-
 ifndef DOCKER
 	DOCKER = docker
 endif