Update Main.java

trendyol-data-eng-summer-intern-2019 · Jul 17, 2019 · eed1557 · eed1557
1 parent dd74c0f
commit eed1557
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/src/main/java/com/trendyol/recomengine/ml/Main.java b/src/main/java/com/trendyol/recomengine/ml/Main.java
@@ -21,15 +21,13 @@
 public class Main {
 
     public static void main(String[] args) {
-        String reviewPath = null;
-        String modelPath = null;
-        if (args.length == 2) {
-            reviewPath = args[0];
-            modelPath = args[1];
-        } else {
+        if (args.length != 2) {
             System.out.println("Expected arguments: <review-path> <model-path>");
             System.exit(0);
         }
+
+        String reviewPath = args[0];
+        String modelPath = args[1];
 
         SparkSession spark = SparkSession
                 .builder()
@@ -152,4 +150,4 @@ public long getTimestamp() {
             return timestamp;
         }
     }
-}
+}