From 815bdd05af7cf288c02de56ea61879ecdd26b2fc Mon Sep 17 00:00:00 2001
From: Eric Hambro <eric.hambro@gmail.com>
Date: Tue, 8 Jun 2021 15:07:59 -0700
Subject: [PATCH] Fixup to reflect Heiner's changes to TB.

---
 nethack_baselines/torchbeast/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nethack_baselines/torchbeast/Dockerfile b/nethack_baselines/torchbeast/Dockerfile
index 13e32b5..eb2897a 100644
--- a/nethack_baselines/torchbeast/Dockerfile
+++ b/nethack_baselines/torchbeast/Dockerfile
@@ -1,7 +1,7 @@
 # -*- mode: dockerfile -*-
 FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
 
-ARG PYTHON_VERSION=3.8
+ARG PYTHON_VERSION=3.7
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && apt-get install -yq \
@@ -43,7 +43,7 @@ RUN python -c 'import torch'
 
 # Install TorchBeast
 WORKDIR /opt/
-RUN git clone https://github.com/condnsdmatters/torchbeast.git --branch eric/experimental-port  --recursive
+RUN git clone https://github.com/facebookresearch/torchbeast.git --recursive
 
 WORKDIR /opt/torchbeast
 RUN pip install -r requirements.txt
-- 
GitLab