Upload input_preprocess.py

input_preprocess.py

@@ -13,7 +13,6 @@ import os
 from tqdm import tqdm
 import time
 import pickle
-import DeepMIMOv3
 
 #%% Scenarios List
 def scenarios_list():
@@ -42,7 +41,7 @@ def tokenizer(selected_scenario_names=None, manual_data=None, gen_raw=True):
 
     if manual_data is not None:
         patches = patch_maker(np.expand_dims(np.array(manual_data), axis=1))
-    else: 
+    else:
         # Patch generation or loading
         deepmimo_data = [DeepMIMO_data_gen(scenario_name) for scenario_name in selected_scenario_names]
         n_scenarios = len(selected_scenario_names)
@@ -122,6 +121,7 @@ def DeepMIMO_data_gen(scenario):
     Returns:
         data (dict): Loaded or generated data.
     """
+    import DeepMIMOv3
     
     parameters, row_column_users, n_ant_bs, n_ant_ue, n_subcarriers = get_parameters(scenario)