From 5da1e5d36d00fbae851cb675f93fb0db44bc6ef7 Mon Sep 17 00:00:00 2001
From: Wil Brady <25513670+WilBrady@users.noreply.github.com>
Date: Wed, 13 Jul 2022 20:24:42 -0400
Subject: [PATCH] Eager mode: Fix some python warnings. (#12167)

---
 orttraining/orttraining/eager/opgen/opgen.py           | 10 ++++++----
 orttraining/orttraining/eager/opgen/opgen/atenops.py   |  5 +----
 .../orttraining/eager/opgen/opgen/custom_ops.py        | 10 +---------
 3 files changed, 8 insertions(+), 17 deletions(-)

diff --git a/orttraining/orttraining/eager/opgen/opgen.py b/orttraining/orttraining/eager/opgen/opgen.py
index 9f90c8e611..ef216e71cc 100755
--- a/orttraining/orttraining/eager/opgen/opgen.py
+++ b/orttraining/orttraining/eager/opgen/opgen.py
@@ -4,13 +4,13 @@
 # Licensed under the MIT License.
 
 import argparse
-import os
 import sys
+import types
 from importlib.machinery import SourceFileLoader
 
-from opgen.generator import ORTGen as ORTGen
+from opgen.generator import ORTGen
 from opgen.parser import cpp_create_from_file as CPPParser
-from opgen.writer import SourceWriter as SourceWriter
+from opgen.writer import SourceWriter
 
 parser = argparse.ArgumentParser(description="Generate ORT ATen operations")
 parser.add_argument(
@@ -23,7 +23,9 @@ parser.add_argument(
 )
 
 args = parser.parse_args()
-ops_module = SourceFileLoader("opgen.customop", args.ops_module).load_module()
+loader = SourceFileLoader("", args.ops_module)
+ops_module = types.ModuleType(loader.name)
+loader.exec_module(ops_module)
 
 ortgen = ORTGen(
     ops_module.ops,
diff --git a/orttraining/orttraining/eager/opgen/opgen/atenops.py b/orttraining/orttraining/eager/opgen/opgen/atenops.py
index 477631ffa4..75efd71f7f 100644
--- a/orttraining/orttraining/eager/opgen/opgen/atenops.py
+++ b/orttraining/orttraining/eager/opgen/opgen/atenops.py
@@ -1,10 +1,7 @@
 from copy import deepcopy
 
 import torch
-from opgen.generator import MakeTorchFallback as MakeTorchFallback
-from opgen.generator import ONNXOp as ONNXOp
-from opgen.generator import ORTGen as ORTGen
-from opgen.generator import SignatureOnly as SignatureOnly
+from opgen.generator import MakeTorchFallback, ONNXOp, SignatureOnly
 from opgen.onnxops import *
 from packaging import version
 
diff --git a/orttraining/orttraining/eager/opgen/opgen/custom_ops.py b/orttraining/orttraining/eager/opgen/opgen/custom_ops.py
index 61815f930f..ab0b11cebf 100644
--- a/orttraining/orttraining/eager/opgen/opgen/custom_ops.py
+++ b/orttraining/orttraining/eager/opgen/opgen/custom_ops.py
@@ -1,12 +1,4 @@
-from copy import deepcopy
-
-from opgen.generator import AttrType
-from opgen.generator import MakeTorchFallback as MakeTorchFallback
-from opgen.generator import ONNXAttr
-from opgen.generator import ONNXOp as ONNXOp
-from opgen.generator import ORTGen as ORTGen
-from opgen.generator import SignatureOnly as SignatureOnly
-from opgen.onnxops import *
+from opgen.onnxops import BatchNormalization, Gemm
 
 ops = {
     "gemm": Gemm("A", "B", "C", "alpha", "beta", "transA", "transB"),