package changes

Dockerfile

@@ -15,6 +15,7 @@ COPY . .
 
 # Create Conda environment from env.yaml
 RUN conda env create -f env.yml
+RUN conda run -n dockformer-venv pip install git+https://github.com/NVIDIA/dllogger.git
 # SHELL ["conda", "run", "-n", "dockformer-venv", "/bin/bash", "-c"]
 
 

env.yml

@@ -29,6 +29,6 @@ dependencies:
   - pip:
 #      - torch==2.3.0 --index-url https://download.pytorch.org/whl/cu118
       - dm-tree==0.1.6
-      - git+https://github.com/NVIDIA/dllogger.git --no-deps
+#      - git+https://github.com/NVIDIA/dllogger.git
       - gradio
       - gradio_molecule3d