From afd5f32e56d3d30bfa98fb35deff51febf50e594 Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Tue, 11 Apr 2023 19:09:36 +0300 Subject: Improve extension --- src/asm_pass1.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/asm_pass1.py') diff --git a/src/asm_pass1.py b/src/asm_pass1.py index fee72ff..d462526 100644 --- a/src/asm_pass1.py +++ b/src/asm_pass1.py @@ -1,4 +1,6 @@ import importlib +import sys +import os.path from dataclasses import dataclass from asm_ops import * from asm_eval import * @@ -33,6 +35,7 @@ def pass_1(result0: Result0) -> Result1: case "@use": module_name, ops = args[0].split(":") ops_split = ops.split(",") + sys.path.append(os.path.dirname(line.src_file)) module = importlib.import_module(module_name) mod_expansions: OpExpansionDict = module.expansions for op in mod_expansions: -- cgit v1.3