From 1d680a1776b626f9105c4bfff11f2dc1c6cdb28e Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 15 Oct 2019 14:18:30 +0100
Subject: [PATCH] [Docker] use correct brevitas branch, fix envvars

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

diff --git a/Dockerfile b/Dockerfile
index e4c682fe5..99f381788 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,7 +4,7 @@ ARG PYTHON_VERSION=3.6
 
 WORKDIR /workspace
 RUN git clone https://github.com/maltanar/brevitas_cnv_lfc.git
-RUN git clone https://github.com/Xilinx/brevitas
+RUN git clone --branch feature/finn_onnx_export https://github.com/Xilinx/brevitas
 RUN cd brevitas; pip install .
 
 COPY requirements.txt .
@@ -15,4 +15,4 @@ RUN rm requirements.txt
 # Note that we expect the cloned finn directory on the host to be
 # mounted on /workspace/finn -- see run-docker.sh for an example
 # of how to do this.
-ENV PYTHONPATH "${PYTHONPATH}:/workspace/finn"
+ENV PYTHONPATH "${PYTHONPATH}:/workspace/finn:/workspace/brevitas_cnv_lfc/training_scripts"
-- 
GitLab