Minor updates (#65)

* parser_utils: fix assertion

otherwise passing -1 (cpu) for emmental device would throw an error

* Update torch requirements

* Update tensorboard version

* Update ujson and pydantic requirements

bootleg/utils/parser/parser_utils.py

@@ -265,9 +265,9 @@ def get_boot_config(config, parser_hierarchy=None, parser=None, unknown=None):
     assert all(
         unknown[idx].startswith(("-", "--")) for idx in range(0, len(unknown), 2)
     )
-    assert all(
-        not unknown[idx].startswith(("-", "--")) for idx in range(1, len(unknown), 2)
-    )
+    for idx in range(1, len(unknown), 2):
+        # allow passing -1 for emmental.device argument
+        assert not unknown[idx].startswith(("-", "--")) or (unknown[idx-1] == "emmental.device" and unknown[idx] == "-1")
     for idx in range(0, len(unknown), 2):
         arg = unknown[idx]
         # If override one you already have in json

setup.py

@@ -18,18 +18,18 @@
         "numpy~=1.19.0",
         "pandas~=1.2.3",
         "progressbar==2.5",
-        "pydantic==1.7.4",
+        "pydantic~=1.7",
         "rich==10.1.0",
         "scikit_learn~=0.24.1",
         "scipy~=1.6.1",
         "sentencepiece==0.1.*",
         "spacy==3.0.*",
         "tagme==0.1.3",
         "tensorboardX==2.1.*",
-        "tensorboard==2.4.*",
-        "torch~=1.7.0",
+        "tensorboard==2.5.*",
+        "torch>=1.7.0,<1.10.0",
         "tqdm>=4.27",
         "transformers>=4.0.0,<5.0.0",
-        "ujson==4.0.2",
+        "ujson~=4.1",
     ],
 )