@@ -728,13 +728,11 @@ def generate_daf_script(self):
728728
729729 accounts : list [ObjectInfo ] = self .lb_accounts .get ()
730730
731- accounts_str , imports , other_str = convert_objects_to_script (accounts )
731+ accounts_str , imports = convert_objects_to_script (accounts )
732732 imports = "\n " .join (set (imports ))
733- if other_str != "" :
734- other_str = "\n " + other_str
735733
736734 if logger_is_present :
737- logger_str , logger_imports , _ = convert_objects_to_script (logger )
735+ logger_str , logger_imports = convert_objects_to_script (logger )
738736 logger_imports = "\n " .join (set (logger_imports ))
739737 else :
740738 logger_imports = ""
@@ -746,7 +744,7 @@ def generate_daf_script(self):
746744 kwargs ["username" ] = connection_mgr .auth .login
747745 kwargs ["password" ] = connection_mgr .auth .password
748746
749- remote_str , remote_imports , _ = convert_objects_to_script (ObjectInfo (daf .RemoteAccessCLIENT , kwargs ))
747+ remote_str , remote_imports = convert_objects_to_script (ObjectInfo (daf .RemoteAccessCLIENT , kwargs ))
750748 remote_imports = "\n " .join (set (remote_imports ))
751749 else :
752750 remote_str , remote_imports = "" , ""
@@ -767,7 +765,7 @@ def generate_daf_script(self):
767765{ remote_imports }
768766{ imports }
769767{ f"from { tracing .__module__ } import { tracing .__class__ .__name__ } " if tracing_is_present else "" }
770- import daf{ other_str }
768+ import daf
771769
772770# Define the logger
773771{ f"logger = { logger_str } " if logger_is_present else "" }
0 commit comments