Skip to content

Commit 1bff774

Browse files
committed
Fixing casatools implicit deps
1 parent a003aa3 commit 1bff774

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

src/samrfi/inference/predictor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from transformers import Sam2Model, Sam2Processor
1414

1515
from samrfi.data import AdaptivePatcher, Preprocessor, SAMDataset
16-
from samrfi.data.ms_loader import MSLoader
1716
from samrfi.utils import logger
1817
from samrfi.utils.errors import CheckpointMismatchError
1918

@@ -537,6 +536,8 @@ def predict_ms(
537536
Returns:
538537
Predicted flags array (baselines, pols, channels, times)
539538
"""
539+
from samrfi.data.ms_loader import MSLoader
540+
540541
logger.info(f"\n{'='*60}")
541542
logger.info("RFI Prediction - Single Pass")
542543
logger.info(f"{'='*60}")
@@ -658,6 +659,8 @@ def predict_iterative(
658659
Returns:
659660
Cumulative flags from all iterations
660661
"""
662+
from samrfi.data.ms_loader import MSLoader
663+
661664
logger.info(f"\n{'='*60}")
662665
logger.info(f"RFI Prediction - Iterative ({num_iterations} passes)")
663666
logger.info(f"{'='*60}")

0 commit comments

Comments
 (0)