diff --git a/tianshou/data/__init__.py b/tianshou/data/__init__.py index d8c5410..c84c2ec 100644 --- a/tianshou/data/__init__.py +++ b/tianshou/data/__init__.py @@ -24,7 +24,13 @@ from tianshou.data.stats import ( SequenceSummaryStats, TimingStats, ) -from tianshou.data.collector import Collector, AsyncCollector, CollectStats, CollectStatsBase, BaseCollector +from tianshou.data.collector import ( + Collector, + AsyncCollector, + CollectStats, + CollectStatsBase, + BaseCollector, +) __all__ = [ "Batch",