Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jul 28, 2021
1 parent 16299b8 commit 24df7b6
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/utilities/test_memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import math

import torch
import torch.nn as nn

Expand Down Expand Up @@ -41,6 +42,7 @@ def test_get_model_size_mb():
# Size will be python version dependent.
assert math.isclose(size_bytes, 0.001319, rel_tol=0.1)


def test_get_sparse_model_size_mb():
class BoringSparseModel(BoringModel):
def __init__(self):
Expand Down

0 comments on commit 24df7b6

Please sign in to comment.