Skip to content

Commit

Permalink
avoid F401 linter error
Browse files Browse the repository at this point in the history
  • Loading branch information
Franck Mamalet committed Oct 9, 2024
1 parent 7918a2b commit f1688dc
Showing 1 changed file with 34 additions and 1 deletion.
35 changes: 34 additions & 1 deletion tests/utils_framework.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
import os
import copy
import math
import numpy as np
from functools import partial
import torch
from torch.nn import Sequential as tSequential
Expand Down Expand Up @@ -76,6 +75,40 @@
from torch.nn import Module as Loss


# to avoid linter F401
__all__ = [
"tLinear",
"tSequential",
"set_seed",
"MeanSquaredError",
"Flatten",
"tSoftmax",
"tMaxPool2d",
"tConv2d",
"tUpSampling2d",
"tConcatenate",
"type_int32",
"GroupSort",
"GroupSort2",
"Sequential",
"FrobeniusLinear",
"FrobeniusConv2d",
"InvertibleDownSampling",
"InvertibleUpSampling",
"evaluate_lip_const",
"DEFAULT_NITER_SPECTRAL_INIT",
"invertible_downsample",
"invertible_upsample",
"bjorck_norm",
"remove_bjorck_norm",
"frobenius_norm",
"remove_frobenius_norm",
"compute_lconv_coef",
"lconv_norm",
"remove_lconv_norm",
"Loss",
]

# not implemented
def module_Unavailable(**kwargs):
return None
Expand Down

0 comments on commit f1688dc

Please sign in to comment.