Update TestCodeFileEditAtomicFlow.py
Browse files
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]
|