diff --git a/deep_data_profiler/classes/torch_profiler.py b/deep_data_profiler/classes/torch_profiler.py
index 0dfc022..ff14844 100644
--- a/deep_data_profiler/classes/torch_profiler.py
+++ b/deep_data_profiler/classes/torch_profiler.py
@@ -6,8 +6,7 @@
import networkx as nx
import torch
from torch.nn.modules import activation, dropout, batchnorm
-from typing import Dict, List, Optional, Tuple, Union
-
+from typing import Dict, List, Optional, Tuple, Union, Any
class TorchProfiler(ABC):
@@ -81,7 +80,7 @@ def __init__(self, model: torch.nn.Module, device: str = "cpu") -> None:
}
self.layerdict = self.create_layers()
- def super_nodes_graph(self) -> Tuple[OrderedDict, nx.DiGraph, Dict]:
+ def super_nodes_graph(self) -> Tuple[OrderedDict, Any, Dict]:
"""
Returns
-------
diff --git a/docs/build/.buildinfo b/docs/build/.buildinfo
index cfb4f15..17be30e 100644
--- a/docs/build/.buildinfo
+++ b/docs/build/.buildinfo
@@ -1,4 +1,4 @@
# Sphinx build info version 1
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: 906fb5efabff5107e6a4b832cbc89a4a
+config: fbd61216107f5aeba0ddba22fc1afcd4
tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/docs/build/.doctrees/algorithms/algorithms.doctree b/docs/build/.doctrees/algorithms/algorithms.doctree
index d8a62e8..6b0b0bd 100644
Binary files a/docs/build/.doctrees/algorithms/algorithms.doctree and b/docs/build/.doctrees/algorithms/algorithms.doctree differ
diff --git a/docs/build/.doctrees/algorithms/modules.doctree b/docs/build/.doctrees/algorithms/modules.doctree
index 0e8d14d..437fac3 100644
Binary files a/docs/build/.doctrees/algorithms/modules.doctree and b/docs/build/.doctrees/algorithms/modules.doctree differ
diff --git a/docs/build/.doctrees/classes/classes.doctree b/docs/build/.doctrees/classes/classes.doctree
index 3b7aed5..fbd2300 100644
Binary files a/docs/build/.doctrees/classes/classes.doctree and b/docs/build/.doctrees/classes/classes.doctree differ
diff --git a/docs/build/.doctrees/classes/modules.doctree b/docs/build/.doctrees/classes/modules.doctree
index 2c36455..419726d 100644
Binary files a/docs/build/.doctrees/classes/modules.doctree and b/docs/build/.doctrees/classes/modules.doctree differ
diff --git a/docs/build/.doctrees/environment.pickle b/docs/build/.doctrees/environment.pickle
index 83d9298..848fe7b 100644
Binary files a/docs/build/.doctrees/environment.pickle and b/docs/build/.doctrees/environment.pickle differ
diff --git a/docs/build/.doctrees/index.doctree b/docs/build/.doctrees/index.doctree
index 60007ed..e2ca800 100644
Binary files a/docs/build/.doctrees/index.doctree and b/docs/build/.doctrees/index.doctree differ
diff --git a/docs/build/.doctrees/install.doctree b/docs/build/.doctrees/install.doctree
index 6c5dea6..865aa4b 100644
Binary files a/docs/build/.doctrees/install.doctree and b/docs/build/.doctrees/install.doctree differ
diff --git a/docs/build/.doctrees/license.doctree b/docs/build/.doctrees/license.doctree
index ede446e..c7238ec 100644
Binary files a/docs/build/.doctrees/license.doctree and b/docs/build/.doctrees/license.doctree differ
diff --git a/docs/build/.doctrees/overview/index.doctree b/docs/build/.doctrees/overview/index.doctree
index 115e5da..35ffb9f 100644
Binary files a/docs/build/.doctrees/overview/index.doctree and b/docs/build/.doctrees/overview/index.doctree differ
diff --git a/docs/build/.doctrees/tableofcontents.doctree b/docs/build/.doctrees/tableofcontents.doctree
index 2f9ffd0..e7bde66 100644
Binary files a/docs/build/.doctrees/tableofcontents.doctree and b/docs/build/.doctrees/tableofcontents.doctree differ
diff --git a/docs/build/_modules/algorithms/homology.html b/docs/build/_modules/algorithms/homology.html
index 04811fb..1414386 100644
--- a/docs/build/_modules/algorithms/homology.html
+++ b/docs/build/_modules/algorithms/homology.html
@@ -3,7 +3,7 @@
- algorithms.homology — Deep Data Profiler 2.0.0 documentation
+ algorithms.homology — Deep Data Profiler 2.0.1 documentation