update np loading for py37 support and add version cli

Author: John Ziegler

analyze.py

@@ -28,7 +28,7 @@
 from torch.autograd import Variable
 from sklearn import metrics
 import traceback
-
+import pkg_resources
 from data.generate_vectors.create_data import create_data
 from main.evaluate_sample import run_eval
 
@@ -38,6 +38,12 @@
 
 def main():
     parser = argparse.ArgumentParser(description="MiMSI Analysis")
+    parser.add_argument(
+        "--version",
+        action="store_true",
+        default=False,
+        help="Display current version of MiMSI",
+    )
     parser.add_argument(
         "--no-cuda",
         action="store_true",
@@ -125,6 +131,10 @@ def main():
     )
 
     args = parser.parse_args()
+    if args.version:
+        print("MiMSI Case Analysis CLI version - " + pkg_resources.require("MiMSI")[0].version)
+        return 
+
     case_list, tumor_bam, normal_bam, case_id, norm_case_id, ms_list, save_loc, cores, saved_model, no_cuda, seed, save, save_format, name, covg, confidence, use_attention = (
         args.case_list,
         args.tumor_bam,

data/data_loader.py

@@ -105,7 +105,7 @@ def __getitem__(self, idx):
 
         sample_id, label = self.__parsefilename__(self.data_file_list[idx])
         final_bags = []
-        bags = self._preprocess(np.load(data_file))
+        bags = self._preprocess(np.load(data_file, allow_pickle=True, encoding="bytes"))
         for bag in bags:
 
             bag = [

data/generate_vectors/create_data.py

@@ -24,6 +24,7 @@
 import warnings
 import pysam
 import traceback
+import pkg_resources
 from glob import glob
 
 from .bam2tensor import Bam2Tensor
@@ -340,6 +341,12 @@ def create_data(
 
 def main():
     parser = argparse.ArgumentParser(description="MiMSI Vector Generation Utility")
+    parser.add_argument(
+        "--version",
+        action="store_true",
+        default=False,
+        help="Display current version of MiMSI",
+    )
 
     single_sample_group = parser.add_argument_group("Single Sample Mode")
     single_sample_group.add_argument(
@@ -389,6 +396,11 @@ def main():
     )
 
     args = parser.parse_args()
+
+    if args.version:
+        print("MiMSI Vector Creation CLI version - " + pkg_resources.require("MiMSI")[0].version)
+        return 
+
     case_list, tumor_bam, normal_bam, case_id, m_list, save_loc, covg, cores, norm_case_id = (
         args.case_list,
         args.tumor_bam,

main/evaluate_sample.py

@@ -22,6 +22,7 @@
 import torch
 import torch.utils.data as data_utils
 import torch.optim as optim
+import pkg_resources
 from torch.autograd import Variable
 from sklearn import metrics
 
@@ -191,6 +192,12 @@ def run_eval(
 
 def main():
     parser = argparse.ArgumentParser(description="MiMSI Sample(s) Evalution Utility")
+    parser.add_argument(
+        "--version",
+        action="store_true",
+        default=False,
+        help="Display current version of MiMSI",
+    )
     parser.add_argument(
         "--no-cuda",
         action="store_true",
@@ -250,6 +257,10 @@ def main():
 
     args = parser.parse_args()
 
+    if args.version:
+        print("MiMSI Case Evaluation Utility version - " + pkg_resources.require("MiMSI")[0].version)
+        return 
+
     saved_model, vector_location, no_cuda, seed, save, save_format, save_loc, name, coverage, confidence, use_attention = (
         args.model,
         args.vector_location,

main/mi_msi_train_test.py

@@ -23,16 +23,23 @@
 import torch
 import torch.utils.data as data_utils
 import torch.optim as optim
+import pkg_resources
 from torch.autograd import Variable
 from sklearn import metrics
 from copy import deepcopy
-from .data.data_loader import MSIBags
-from .model.mi_msi_model import MSIModel
+from data.data_loader import MSIBags
+from model.mi_msi_model import MSIModel
 
 
 def main():
     parser = argparse.ArgumentParser(
         description='MiMSI - A Multiple Instance Learning Model for detecting microsatellite instability in NGS data')
+    parser.add_argument(
+        "--version",
+        action="store_true",
+        default=False,
+        help="Display current version of MiMSI",
+    )
     parser.add_argument('--epochs', type=int, default=40, metavar='N',
                         help='Number of epochs to train (default: 40)')
     parser.add_argument('--lr', type=float, default=0.0001, metavar='LR',
@@ -54,6 +61,11 @@ def main():
 
 
     args = parser.parse_args()
+
+    if args.version:
+        print("MiMSI Model Training & Testing CLI version - " + pkg_resources.require("MiMSI")[0].version)
+        return 
+
     # Training settings
     epochs, lr, reg, seed, name, train_location, test_location, save = args.epochs, args.lr, args.reg, args.seed, args.name, args.train_location, args.test_location, args.save
     cuda = not args.no_cuda and torch.cuda.is_available()

requirements.txt

@@ -1,6 +1,6 @@
 future; python_version <= '2.7'
 nose>=1.3.7
-numpy==1.16.1
+numpy==1.16.5
 pandas==0.24.2; python_version <= '2.7'
 pandas
 pillow>=8.1.1; python_version >= '3.6'