File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1313from transformers import Sam2Model , Sam2Processor
1414
1515from samrfi .data import AdaptivePatcher , Preprocessor , SAMDataset
16- from samrfi .data .ms_loader import MSLoader
1716from samrfi .utils import logger
1817from 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 } " )
You can’t perform that action at this time.
0 commit comments