| 528 | if os.path.isdir(exp_data_dir) == False: |
| 529 | os.mkdir(exp_data_dir) |
| 530 | class SaveExpData: |
| 531 | @classmethod |
| 532 | def INPUT_TYPES(s): |
| 533 | return {"required": { |
| 534 | "file_name": ("STRING", {"multiline": False, "default": ""}), |
| 535 | }, |
| 536 | "optional": {"save_exp": ("EXP_DATA",), } |
| 537 | } |
| 538 | |
| 539 | RETURN_TYPES = ("STRING",) |
| 540 | RETURN_NAMES = ("file_name",) |
| 541 | FUNCTION = "run" |
| 542 | CATEGORY = "AdvancedLivePortrait" |
| 543 | OUTPUT_NODE = True |
| 544 | |
| 545 | def run(self, file_name, save_exp:ExpressionSet=None): |
| 546 | if save_exp == None or file_name == "": |
| 547 | return file_name |
| 548 | |
| 549 | with open(os.path.join(exp_data_dir, file_name + ".exp"), "wb") as f: |
| 550 | dill.dump(save_exp, f) |
| 551 | |
| 552 | return file_name |
| 553 | |
| 554 | class LoadExpData: |
| 555 | @classmethod |
nothing calls this directly
no outgoing calls
no test coverage detected