From fc7dd1dd52ec9da57b9f333d834ed4bf2ff5348b Mon Sep 17 00:00:00 2001 From: alex-crlhmmr Date: Thu, 9 Jul 2026 10:00:46 -0700 Subject: [PATCH] Fix ModuleNotFoundError: unify Lightning imports on pytorch_lightning pretrain.py, classification.py, and segmentation.py import seed_everything from lightning.pytorch, but environment.yml installs pytorch-lightning (not the separate `lightning` distribution), so `conda env create -f environment.yml` followed by a training run fails with `ModuleNotFoundError: No module named 'lightning'`. pytorch_lightning re-exports seed_everything and the rest of the codebase already uses it. --- classification.py | 2 +- pretrain.py | 2 +- segmentation.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/classification.py b/classification.py index 9ba2907..c923f07 100644 --- a/classification.py +++ b/classification.py @@ -96,7 +96,7 @@ def _format_results(checkpoint, results): def main(): args = build_parser().parse_args() import torch - from lightning.pytorch import seed_everything + from pytorch_lightning import seed_everything from utils.training import ( build_trainer, create_run_paths, diff --git a/pretrain.py b/pretrain.py index 372459d..0460c29 100644 --- a/pretrain.py +++ b/pretrain.py @@ -99,7 +99,7 @@ def main(): args = build_parser().parse_args() _validate_sampling_args(args) import torch - from lightning.pytorch import seed_everything + from pytorch_lightning import seed_everything from utils.training import ( build_trainer, create_run_paths, diff --git a/segmentation.py b/segmentation.py index 3ec0d96..32078aa 100644 --- a/segmentation.py +++ b/segmentation.py @@ -111,7 +111,7 @@ def _format_results(root_checkpoint, results_list): def main(): args = build_parser().parse_args() import torch - from lightning.pytorch import seed_everything + from pytorch_lightning import seed_everything from utils.training import ( build_trainer, create_run_paths,