Fixed linting issues

Author: ngoodger

nle_language_wrapper/agents/sample_factory/env.py

@@ -44,7 +44,7 @@ def _tokenize(self, str_obsv):
             max_length=self.cfg["max_token_length"],
         )
         # Sample factory insists on normalizing obs key.
-        tokens.data["obs"] = torch.tensor(0)
+        tokens.data["obs"] = torch.zeros(1)
         return tokens.data
 
     def _convert_obsv_to_str(self, obsv):

pyproject.toml

@@ -17,6 +17,7 @@ known_third_party = ["nle"]
 
 [tool.pylint.messages_control]
 max-line-length = 88
+generated-members="torch.*"
 disable = [
   "missing-function-docstring",
   "missing-module-docstring",

setup.py

@@ -62,6 +62,7 @@ def build_extension(self, ext):
     "dev": [
         "black>=22.6.0",
         "flake8>=4.0.1",
+        "pylint>=2.15.8",
         "pytest>=7.1.2",
         "pytest-cov>=3.0.0",
         "pytest-mock>=3.7.0",
@@ -72,7 +73,8 @@ def build_extension(self, ext):
     "agent": [
         "sample_factory>=1.121.4",
         "transformers>=4.17.0",
-        "torch@https://download.pytorch.org/whl/cu111/torch-1.9.1%2Bcu111-cp39-cp39-linux_x86_64.whl",
+        "torch@https://download.pytorch.org/whl/cu111/"
+        "torch-1.9.1%2Bcu111-cp39-cp39-linux_x86_64.whl",
     ],
 }