Tachi67 commited on
Commit
b28c869
·
1 Parent(s): 589a875

Update TestCodeFileEditAtomicFlow.py

Browse files
Files changed (1) hide show
  1. TestCodeFileEditAtomicFlow.py +1 -0
TestCodeFileEditAtomicFlow.py CHANGED
@@ -3,6 +3,7 @@ from typing import Dict, Any
3
  from flow_modules.Tachi67.CodeFileEditFlowModule import CodeFileEditAtomicFlow
4
 
5
  class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
 
6
  def _generate_import_statement(self, code_lib_location):
7
  module_dir = os.path.dirname(code_lib_location)
8
  module_name = os.path.splitext(os.path.basename(code_lib_location))[0]
 
3
  from flow_modules.Tachi67.CodeFileEditFlowModule import CodeFileEditAtomicFlow
4
 
5
  class TestCodeFileEditAtomicFlow(CodeFileEditAtomicFlow):
6
+ """Refer to: https://huggingface.co/Tachi67/CodeFileEditFlowModule/tree/main"""
7
  def _generate_import_statement(self, code_lib_location):
8
  module_dir = os.path.dirname(code_lib_location)
9
  module_name = os.path.splitext(os.path.basename(code_lib_location))[0]