zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 09:21: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.
import unittest
import inspect
import unittest
import mlx.core as mx
import mlx.optimizers as opt
import mlx.utils
import mlx_tests
def get_all_optimizers():
classes = dict()
for name, obj in inspect.getmembers(opt):
@ -16,8 +17,10 @@ def get_all_optimizers():
classes[name] = obj
return classes
optimizers_dict = get_all_optimizers()
class TestOptimizers(mlx_tests.MLXTestCase):
def test_optimizers(self):
params = {