From 40d63c89c4ef6d1035029e77b7d04a5e9c16d2f3 Mon Sep 17 00:00:00 2001 From: 0x00-pl <0x00.pl@gmail.com> Date: Mon, 26 Aug 2024 00:33:31 +0800 Subject: [PATCH] update code --- plai/core/core_dialect.py | 2 +- plai/core/location.py | 1 - plai/core/module.py | 2 +- 3 files changed, 2 insertions(+), 3 deletions(-) diff --git a/plai/core/core_dialect.py b/plai/core/core_dialect.py index 0e3cba0..a103857 100644 --- a/plai/core/core_dialect.py +++ b/plai/core/core_dialect.py @@ -23,7 +23,7 @@ def build(op_name: str, args: list, attrs: dict, loc: Location = None): class Transpose(CoreNode): - def __init__(self, arg: module.Node, loc: Location=None): + def __init__(self, arg: module.Node, loc: Location = None): super().__init__([arg], {}, loc) @staticmethod diff --git a/plai/core/location.py b/plai/core/location.py index 38e8333..734fcc7 100644 --- a/plai/core/location.py +++ b/plai/core/location.py @@ -1,4 +1,3 @@ - class Location: pass diff --git a/plai/core/module.py b/plai/core/module.py index 5d17a10..fd3d5eb 100644 --- a/plai/core/module.py +++ b/plai/core/module.py @@ -1,5 +1,5 @@ from abc import ABC, abstractmethod -from typing import List, Dict, Optional, Self +from typing import List, Dict, Optional from plai.core.location import Location