ImportError: cannot import name 'GWGenerator' from 'dragonnfruit.io'

[aron0093]

No description provided.

[aron0093]

I think the name was changed to GenomewideGenerator

[jmschrei]

Where is this issue arising?

[aron0093]

I was following the code snippet in the README. Changing to GenomewideGenerator fixed the import error but I haven't run the code yet.

from dragonnfruit.io import GWGenerator

.
.
.

X = torch.utils.data.DataLoader(
	GWGenerator(
		sequence=sequence,
		signal=signal,
		neighbors=neighbors,
		cell_states=cell_states,
		read_depths=read_depths,
		trimming=(2114 - 1000) // 2, 
		window=1000, 
		chroms=['chr1'],
		random_state=None),
	pin_memory=True, 
	num_workers=8,
	worker_init_fn=lambda x: numpy.random.seed(x),
	batch_size=128)