Include missed formatting fixes.
@@ -44,7 +44,8 @@ def generate_samples(net: BayesNet,
return _samples_generator(sample_iterator, vertices_unwrapped)
-def _samples_generator(sample_iterator: Any, vertices_unwrapped : Any) -> Generator[Dict[Tuple[Any, ...], numpy_types], None, None]:
+def _samples_generator(sample_iterator: Any,
+ vertices_unwrapped : Any) -> Generator[Dict[Tuple[Any, ...], numpy_types], None, None]:
while (True):
network_sample = sample_iterator.next()
sample = {Vertex._get_python_id(vertex_unwrapped): Tensor._to_ndarray(network_sample.get(vertex_unwrapped)) for vertex_unwrapped in vertices_unwrapped}
