zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Pr template (#99)

Browse Source 
* pr template
* format fix
This commit is contained in:
Awni Hannun 2023-12-09 09:36:56 -08:00 committed by GitHub
parent fd836d891b
commit 89b90dcfec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/pull_request_template.md vendored Normal file
View File

@ -0,0 +1,4 @@
Before submitting this PR, check the [contribution guidelines](CONTRIBUTING.md).
Make sure your code is formatted: `pre-commit run --all-files`.

5
python/tests/test_optimizers.py
View File

@ -1,13 +1,14 @@
# Copyright © 2023 Apple Inc. # Copyright © 2023 Apple Inc.
import unittest
import inspect import inspect
import unittest
import mlx.core as mx import mlx.core as mx
import mlx.optimizers as opt import mlx.optimizers as opt
import mlx.utils import mlx.utils
import mlx_tests import mlx_tests
def get_all_optimizers(): def get_all_optimizers():
classes = dict() classes = dict()
for name, obj in inspect.getmembers(opt): for name, obj in inspect.getmembers(opt):
@ -16,8 +17,10 @@ def get_all_optimizers():
classes[name] = obj classes[name] = obj
return classes return classes
optimizers_dict = get_all_optimizers() optimizers_dict = get_all_optimizers()
class TestOptimizers(mlx_tests.MLXTestCase): class TestOptimizers(mlx_tests.MLXTestCase):
def test_optimizers(self): def test_optimizers(self):
params = { params = {