[TRAX] Begin pylinting PRs

PiperOrigin-RevId: 340346172

.travis.yml

@@ -16,3 +16,4 @@ install:
   - ./oss_scripts/oss_pip_install.sh
 script:
   - ./oss_scripts/oss_tests.sh
+  - pylint trax;

pylintrc

@@ -37,7 +37,72 @@ msg-template={msg_id}:{line:3} {obj}: {msg} [{symbol}]
 enable=indexing-exception,old-raise-syntax
 
 # List of checkers and warnings to disable.
-disable=design,similarities,no-self-use,attribute-defined-outside-init,locally-disabled,star-args,pointless-except,bad-option-value,global-statement,fixme,suppressed-message,useless-suppression,locally-enabled,file-ignored,multiple-imports,c-extension-no-member,trailing-newlines,unsubscriptable-object,misplaced-comparison-constant,no-member,abstract-method,no-else-return,missing-docstring,wrong-import-order,protected-access,inconsistent-return-statements,invalid-unary-operand-type,import-error,no-name-in-module,arguments-differ,not-context-manager,unused-argument
+disable=abstract-method,
+	arguments-differ,
+	attribute-defined-outside-init,
+	attribute-defined-outside-init,
+	bad-option-value,
+	bad-option-value,
+	c-extension-no-member,
+	c-extension-no-member,
+	design,
+	design,
+	file-ignored,
+	file-ignored,
+	fixme,
+	fixme,
+	g-no-augmented-assignment,
+	global-statement,
+	global-statement,
+	implicit-str-concat-in-sequence,
+	import-error,
+	import-outside-toplevel,
+	inconsistent-return-statements,
+	invalid-metaclass,
+	invalid-unary-operand-type,
+	locally-disabled,
+	locally-disabled,
+	locally-enabled,
+	locally-enabled,
+	misplaced-comparison-constant,
+	misplaced-comparison-constant,
+	missing-docstring,
+	multiple-imports,
+	multiple-imports,
+	no-else-break,
+	no-else-continue,
+	no-else-raise,
+	no-else-return,
+	no-else-return,
+	no-member,
+	no-name-in-module,
+	no-self-use,
+	no-self-use,
+	not-context-manager,
+	pointless-except,
+	pointless-except,
+	protected-access,
+	similarities,
+	similarities,
+	star-args,
+	star-args,
+	suppressed-message,
+	suppressed-message,
+	trailing-newlines,
+	trailing-newlines,
+	ungrouped-imports,
+	unnecessary-pass,
+	unsubscriptable-object,
+	unsubscriptable-object,
+	unused-argument,
+	useless-else-on-loop,
+	useless-object-inheritance,
+	useless-suppression,
+	useless-suppression,
+	wrong-import-order,
+	wrong-import-order,
+	wrong-import-position,
+
 
 [BASIC]
 

setup.py

@@ -42,6 +42,7 @@
         'jax',
         'jaxlib',
         'numpy',
+        'psutil',
         'scipy',
         'six',
         't5',