diff --git a/.github/workflows/test-matrix.yml b/.github/workflows/test-matrix.yml index e910f745..18339f01 100644 --- a/.github/workflows/test-matrix.yml +++ b/.github/workflows/test-matrix.yml @@ -3,11 +3,7 @@ name: Python application -on: - push: - branches: [ master ] - pull_request: - branches: [ master ] +on: [push, pull_request] jobs: diff --git a/docs/tensorboard.rst b/docs/tensorboard.rst index 99f03ea3..75e8377a 100644 --- a/docs/tensorboard.rst +++ b/docs/tensorboard.rst @@ -11,8 +11,3 @@ tensorboardX :members: .. automethod:: __init__ - -.. autoclass:: TorchVis - :members: - - .. automethod:: __init__ \ No newline at end of file diff --git a/tensorboardX/__init__.py b/tensorboardX/__init__.py index 51812471..4244d392 100644 --- a/tensorboardX/__init__.py +++ b/tensorboardX/__init__.py @@ -3,7 +3,6 @@ from .global_writer import GlobalSummaryWriter from .record_writer import RecordWriter -from .torchvis import TorchVis from .writer import FileWriter, SummaryWriter try: diff --git a/tensorboardX/torchvis.py b/tensorboardX/torchvis.py deleted file mode 100644 index 735abfec..00000000 --- a/tensorboardX/torchvis.py +++ /dev/null @@ -1,47 +0,0 @@ -import gc - -from .writer import SummaryWriter - -# Supports TensorBoard visualization -vis_formats = {'tensorboard': SummaryWriter} - - -class TorchVis: - def __init__(self, *args, **init_kwargs): - """ - Args: - args (list of strings): The name of the visualization target(s). - Accepted targets are 'tensorboard'. - init_kwargs: Additional keyword parameters for the writer. - """ - self.subscribers = {} - self.register(*args, **init_kwargs) - - def register(self, *args, **init_kwargs): - # Sets tensorboard as the default visualization format if not specified - formats = args if args else ['tensorboard'] - for format in formats: - if self.subscribers.get(format) is None and format in vis_formats: - self.subscribers[format] = vis_formats[format](**init_kwargs.get(format, {})) - - def unregister(self, *args): - for format in args: - if format in self.subscribers: - self.subscribers[format].close() - del self.subscribers[format] - gc.collect() - - def __getattr__(self, attr): - if not self.subscribers: - raise AttributeError(f"'{type(self).__name__}' object has no attribute '{attr}'") - - def wrapper(*args, **kwargs): - for _, subscriber in self.subscribers.items(): - if hasattr(subscriber, attr): - getattr(subscriber, attr)(*args, **kwargs) - return wrapper - - # Handle writer management (open/close) for the user - def __del__(self): - for _, subscriber in self.subscribers.items(): - subscriber.close()