fix: remove non-existent WindowGenerator import from data/__init__.py

andrewkern · andrewkern · commit 8905318cf198 · 2025-07-06T22:30:13.000-07:00
- WindowGenerator class does not exist in windows.py
- Only generate_genomic_windows function is available
- This fixes ImportError in all test files
diff --git a/locator/data/__init__.py b/locator/data/__init__.py
@@ -11,7 +11,7 @@
 )
 from .indexset import IndexSet
 from .tf_dataset import flip_genotypes_tf, make_tf_dataset, make_tf_dataset_from_arrays
-from .windows import WindowGenerator, generate_genomic_windows
+from .windows import generate_genomic_windows
 
 __all__ = [
     "FilterStats",
@@ -25,6 +25,5 @@
     "make_tf_dataset",
     "make_tf_dataset_from_arrays",
     "flip_genotypes_tf",
-    "WindowGenerator",
     "generate_genomic_windows",
 ]

Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,7 @@`
`11`	`11`	`)`
`12`	`12`	`from .indexset import IndexSet`
`13`	`13`	`from .tf_dataset import flip_genotypes_tf, make_tf_dataset, make_tf_dataset_from_arrays`
`14`		`-from .windows import WindowGenerator, generate_genomic_windows`
	`14`	`+from .windows import generate_genomic_windows`
`15`	`15`
`16`	`16`	`__all__ = [`
`17`	`17`	`"FilterStats",`
`@@ -25,6 +25,5 @@`
`25`	`25`	`"make_tf_dataset",`
`26`	`26`	`"make_tf_dataset_from_arrays",`
`27`	`27`	`"flip_genotypes_tf",`
`28`		`- "WindowGenerator",`
`29`	`28`	`"generate_genomic_windows",`
`30`	`29`	`]`