search

google / jetstream-pytorch

PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
Apache License 2.0
33 stars 14 forks source link

Cleanup test_jax_torch #48

Closed FanhaiLu1 closed 4 months ago

FanhaiLu1 commented 4 months ago

After this PR: Your code has been rated at 10.00/10

Before this PR: Your code has been rated at 8.95/10 ***** Module tests.test_jax_torch test_jax_torch.py:9:0: C0115: Missing class docstring (missing-class-docstring) test_jax_torch.py:11:2: C0116: Missing function or method docstring (missing-function-docstring) test_jax_torch.py:30:2: C0116: Missing function or method docstring (missing-function-docstring) test_jax_torch.py:6:0: C0411: standard import "unittest" should be placed before third party imports "torch", "torch_xla2", "jax", "jax.numpy" (wrong-import-order)