From 27042f9f5c17b70531c464b3f61178050d636d24 Mon Sep 17 00:00:00 2001
From: yoels <yoels@login-a.zuvela.ibm.com>
Date: Sun, 22 Dec 2024 09:27:18 +0100
Subject: [PATCH] fixed imports

---
 fuse/data/__init__.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/fuse/data/__init__.py b/fuse/data/__init__.py
index 6c192030..74206ad2 100644
--- a/fuse/data/__init__.py
+++ b/fuse/data/__init__.py
@@ -8,13 +8,23 @@
     OpFunc,
     OpKeepKeypaths,
     OpLambda,
+    OpRepeat,
     OpSet,
 )
 from fuse.data.ops.ops_read import OpReadDataframe
 from fuse.data.pipelines.pipeline_default import PipelineDefault
 from fuse.data.utils.collates import CollateDefault
 from fuse.data.utils.export import ExportDataset
-from fuse.utils.ndict import NDict  # Fix circular import
-from fuse.version import __version__
 
 # import shortcuts
+from fuse.data.utils.sample import (
+    create_initial_sample,
+    get_initial_sample_id,
+    get_initial_sample_id_key,
+    get_sample_id,
+    get_sample_id_key,
+    get_specific_sample_from_potentially_morphed,
+    set_sample_id,
+)
+from fuse.utils.ndict import NDict  # Fix circular import
+from fuse.version import __version__