better management of wandb key

Author: amacrutherford

Dockerfile

@@ -31,6 +31,5 @@ RUN export TF_FORCE_GPU_ALLOW_GROWTH=true
 # RUN pip install jupyterlab
 
 #for secrets and debug
-ENV WANDB_API_KEY=""
 ENV WANDB_ENTITY=""
 RUN git config --global --add safe.directory /home/${MYUSER}

Makefile

@@ -9,14 +9,15 @@ endif
 
 # Set flag for docker run command
 MYUSER=myuser
+WANDB_API_KEY=$(shell cat ./wandb_key)
 BASE_FLAGS=-it --rm -v ${PWD}:/home/$(MYUSER) --shm-size 20G
-RUN_FLAGS=$(GPUS) $(BASE_FLAGS)
+RUN_FLAGS=$(GPUS) $(BASE_FLAGS) -e WANDB_API_KEY=$(WANDB_API_KEY)
 
-DOCKER_IMAGE_NAME = jaxmarl
-IMAGE = $(DOCKER_IMAGE_NAME):latest
+DOCKER_IMAGE_NAME=jaxmarl
+IMAGE=$(DOCKER_IMAGE_NAME):latest
 DOCKER_RUN=docker run $(RUN_FLAGS) $(IMAGE)
-USE_CUDA = $(if $(GPUS),true,false)
-ID = $(shell id -u)
+USE_CUDA=$(if $(GPUS),true,false)
+ID=$(shell id -u)
 
 # make file commands
 build: