[BUGFIX] Fix an import bug in `data_pipeline.py`

2022-10-18 20:08:14 -04:00 · 2022-10-18 20:08:14 -04:00 · 7f84eebd48
parent a51f5fb585
commit 7f84eebd48
1 changed files with 1 additions and 0 deletions
--- a/openfold/data/data_pipeline.py
+++ b/openfold/data/data_pipeline.py
@ -19,6 +19,7 @@ from multiprocessing import cpu_count
 from typing import Mapping, Optional, Sequence, Any

 import numpy as np
+import torch

 from openfold.data import templates, parsers, mmcif_parsing
 from openfold.data.templates import get_custom_template_features