diff --git a/src/dsdk/__init__.py b/src/dsdk/__init__.py
index 703e981..c3fa4f9 100644
--- a/src/dsdk/__init__.py
+++ b/src/dsdk/__init__.py
@@ -22,6 +22,7 @@
     dump_pickle_file,
     load_json_file,
     load_pickle_file,
+    profile,
     retry,
 )