Skip to content

Conversation

@preshanth
Copy link
Owner

The goal here is more more general RFI generation code into rfi toolbox and it has been done. I think this branch is good to go into main. Once tested.

  SAM-RFI now depends on rfi_toolbox for shared utilities.

  ## Changes

  **src/samrfi/data/__init__.py**:
  - Forward import MSLoader, Preprocessor, GPUPreprocessor from rfi_toolbox
  - Forward import TorchDataset, BatchWriter from rfi_toolbox.datasets
  - Keep SAM2-specific modules (SAMDataset, AdaptivePatcher, GPU utilities)

  **src/samrfi/evaluation/__init__.py**:
  - Forward import all metrics from rfi_toolbox.evaluation
  - Forward import inject_synthetic_data from rfi_toolbox.io
  - All core metrics now shared via rfi_toolbox

  **src/samrfi/data_generation/__init__.py**:
  - Forward import SyntheticDataGenerator from rfi_toolbox
  - Keep MSDataGenerator (SAM2-specific)

  **pyproject.toml**:
  - Added note about rfi_toolbox dependency

  ## Testing

  ✓ All imports work: MSLoader, Preprocessor, metrics, SyntheticDataGenerator
  ✓ Backward compatibility maintained via forward imports
  ✓ No breaking changes for existing SAM-RFI code

  ## Status

  - rfi_toolbox: Phases 1-3 complete (commit 6d8d29f)
  - SAM-RFI: Phase 4 complete (this commit)
@preshanth preshanth requested a review from Kitchi December 31, 2025 09:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants