@@ -249,6 +249,8 @@ def query_cache(self):
249249 self ._is_cached_valid = False
250250
251251 def copy_code (self , output_dir ):
252+ from sebs .faas .function import Language
253+
252254 FILES = {
253255 "python" : ["*.py" , "requirements.txt*" ],
254256 "nodejs" : ["*.js" , "package.json" ],
@@ -261,7 +263,7 @@ def copy_code(self, output_dir):
261263 shutil .copy2 (os .path .join (path , f ), output_dir )
262264
263265 # copy src folder of java (java benchmarks are maven project and need directories)
264- if self .language_name == "java" :
266+ if self .language == Language . JAVA :
265267 output_src_dir = os .path .join (output_dir , "src" )
266268
267269 if os .path .exists (output_src_dir ):
@@ -277,8 +279,8 @@ def copy_code(self, output_dir):
277279
278280 #This is for making jar file and add it to docker directory
279281 def add_java_output (self , code_dir ):
280-
281- if self .language_name == "java" :
282+ from sebs . faas . function import Language
283+ if self .language == Language . JAVA :
282284
283285 # Step 1: Move Main.java o src directory
284286 src_dir = os .path .join (code_dir , "src" , "main" , "java" )
0 commit comments