transe in package

conema · Jan 14, 2021 · 02bc722 · 02bc722
1 parent 2e7e0dd
commit 02bc722
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 5 deletions.
diff --git a/Embedding.py → TransEmodule/Embedding.py b/Embedding.py → TransEmodule/Embedding.py
diff --git a/TransE.py → TransEmodule/TransE.py b/TransE.py → TransEmodule/TransE.py
@@ -2,8 +2,8 @@
 from math import sqrt, ceil
 import random
 import time
-from Embedding import Embedding
-import utils
+from TransEmodule.Embedding import Embedding
+from TransEmodule import utils
 
 
 class TransE:

diff --git a/TransEmodule/__init__.py b/TransEmodule/__init__.py
diff --git a/utils.py → TransEmodule/utils.py b/utils.py → TransEmodule/utils.py
diff --git a/example.py b/example.py
@@ -1,8 +1,8 @@
 import findspark
 from pyspark.sql import SparkSession
 from pyspark import SparkContext, SparkConf
-import utils
-from TransE import TransE
+from TransEmodule import utils
+from TransEmodule.TransE import TransE
 
 findspark.init()
 

diff --git a/test.py b/test.py
@@ -1,7 +1,7 @@
 import numpy as np
 from pyspark.sql import SparkSession
 from pyspark import SparkContext, SparkConf
-import utils
+from TransEmodule import utils
 
 
 def check_entities(x, map):