Cast header values to integers

src/preprocessing.py

@@ -1,11 +1,10 @@
-from sys import argv
 from pandas import read_table
 
 
 def read_header(filename):
     with open(filename, "r") as f:
         header = f.readline().split()
-    return header[0], header[1]
+    return int(header[0]), int(header[1])
 
 
 def parse_file(filename):
@@ -14,11 +13,3 @@ def parse_file(filename):
         filename, names=["source", "destination", "distance"], sep=" ", skiprows=[0]
     )
     return n, m, df
-
-
-def main():
-    parse_file(argv[1])
-
-
-if __name__ == "__main__":
-    main()