Skip to content

GitLab

Commit 5e341f60 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 60e9bd54
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
openfold/data/data_modules.py
View file @ 5e341f60
...@@ -617,7 +617,6 @@ class OpenFoldDataModule(pl.LightningDataModule): ...@@ -617,7 +617,6 @@ class OpenFoldDataModule(pl.LightningDataModule):
if(self.training_mode): if(self.training_mode):
train_dataset = dataset_gen( train_dataset = dataset_gen(
data_dir=self.train_data_dir, data_dir=self.train_data_dir,
chain_data_cache_path=self.train_data_cache_path,
alignment_dir=self.train_alignment_dir, alignment_dir=self.train_alignment_dir,
filter_path=self.train_filter_path, filter_path=self.train_filter_path,
max_template_hits=self.config.train.max_template_hits, max_template_hits=self.config.train.max_template_hits,
...@@ -632,7 +631,6 @@ class OpenFoldDataModule(pl.LightningDataModule): ...@@ -632,7 +631,6 @@ class OpenFoldDataModule(pl.LightningDataModule):
if(self.distillation_data_dir is not None): if(self.distillation_data_dir is not None):
distillation_dataset = dataset_gen( distillation_dataset = dataset_gen(
data_dir=self.distillation_data_dir, data_dir=self.distillation_data_dir,
chain_data_cache_path=self.distillation_chain_data_cache_path,
alignment_dir=self.distillation_alignment_dir, alignment_dir=self.distillation_alignment_dir,
filter_path=self.distillation_filter_path, filter_path=self.distillation_filter_path,
max_template_hits=self.config.train.max_template_hits, max_template_hits=self.config.train.max_template_hits,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment