Clean-up imports

pyg-team · Jul 17, 2023 · 4f372c7 · 4f372c7
1 parent ee13f98
commit 4f372c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/benchmark/utils/__init__.py b/benchmark/utils/__init__.py
@@ -4,7 +4,6 @@
 from .utils import get_split_masks
 from .utils import save_benchmark_data, write_to_csv
 from .utils import test
-from .utils import xpu_profiler
 
 __all__ = [
     'emit_itt',
@@ -15,5 +14,4 @@
     'save_benchmark_data',
     'write_to_csv',
     'test',
-    'xpu_profiler',
 ]
diff --git a/torch_geometric/profile/__init__.py b/torch_geometric/profile/__init__.py
@@ -4,6 +4,7 @@
     print_time_total,
     rename_profile_file,
     torch_profile,
+    xpu_profile,
 )
 from .utils import count_parameters
 from .utils import get_model_size
@@ -21,6 +22,7 @@
     'print_time_total',
     'rename_profile_file',
     'torch_profile',
+    'xpu_profile',
     'count_parameters',
     'get_model_size',
     'get_data_size',