From 78aa7a38af5f12b469a04b4d1aedd8127aafd656 Mon Sep 17 00:00:00 2001 From: sichu Date: Thu, 19 Dec 2024 21:59:37 +0000 Subject: [PATCH] ruff --- sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py | 1 - 1 file changed, 1 deletion(-) diff --git a/sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py b/sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py index 037bd2e5d..175e9c00d 100644 --- a/sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py +++ b/sub-packages/bionemo-esm2/src/bionemo/esm2/scripts/train_esm2.py @@ -30,7 +30,6 @@ from bionemo.esm2.data.datamodule import ESMDataModule from bionemo.esm2.data.dataset import RandomMaskStrategy from bionemo.esm2.data.tokenizer import get_tokenizer -from bionemo.llm.lightning import PerplexityLoggingCallback from bionemo.llm.model.biobert.lightning import biobert_lightning_module from bionemo.llm.model.biobert.model import BiobertSpecOption from bionemo.llm.model.lr_scheduler import WarmupAnnealDecayHoldScheduler