Skip to content

Commit a0ebd6b

Browse files
author
Rohith Krishna
committed
format
1 parent 7707677 commit a0ebd6b

File tree

3 files changed

+6
-5
lines changed

3 files changed

+6
-5
lines changed

src/modelhub/cli.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ def fold(inputs: str):
1717
cfg = compose(config_name="inference", overrides=overrides)
1818
run_inference(cfg)
1919

20+
2021
@app.command()
2122
def predict(inputs: str):
2223
"""Alias for fold command."""

src/modelhub/data/pipelines.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -284,10 +284,10 @@ def build_af3_transform_pipeline(
284284
),
285285
},
286286
),
287-
#NOTE: this is used in training to pad DNA sequences, but we don't use it in inference
288-
#TrainingRoute(
289-
#PadDNA(p_skip=pad_dna_p_skip),
290-
#),
287+
# NOTE: this is used in training to pad DNA sequences, but we don't use it in inference
288+
# TrainingRoute(
289+
# PadDNA(p_skip=pad_dna_p_skip),
290+
# ),
291291
FlagAndReassignCovalentModifications(),
292292
FlagNonPolymersForAtomization(),
293293
]

src/modelhub/inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,4 +44,4 @@ def run_inference(cfg: DictConfig) -> None:
4444

4545

4646
if __name__ == "__main__":
47-
run_inference()
47+
run_inference()

0 commit comments

Comments
 (0)