Skip to content

Commit

Permalink
Fixed missing superclass
Browse files Browse the repository at this point in the history
  • Loading branch information
Eixalde committed Jun 15, 2023
1 parent a0bfdec commit 5f044b3
Showing 1 changed file with 17 additions and 17 deletions.
34 changes: 17 additions & 17 deletions vip_hci/var/paramenum.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from enum import Enum


class SvdMode(Enum):
class SvdMode(str, Enum):
"""
Define the various modes to use with SVD in PCA as constant strings.
Expand Down Expand Up @@ -54,7 +54,7 @@ class SvdMode(Enum):
RANDPYTORCH = "randpytorch"


class Scaling(Enum):
class Scaling(str, Enum):
"""
Define modes for the pixel-wise scaling.
Expand All @@ -77,7 +77,7 @@ class Scaling(Enum):
SPATSTANDARD = "spat-standard"


class Adimsdi(Enum):
class Adimsdi(str, Enum):
"""
Define modes for processing ADI+mSDI cubes through PCA.
Expand All @@ -101,7 +101,7 @@ class Adimsdi(Enum):


# TODO: document all modes
class Imlib(Enum):
class Imlib(str, Enum):
"""
Define modes for image transformations to be used.
Expand All @@ -123,7 +123,7 @@ class Imlib(Enum):


# TODO: document all modes
class Interpolation(Enum):
class Interpolation(str, Enum):
"""
Define modes for interpolation.
Expand Down Expand Up @@ -154,7 +154,7 @@ class Interpolation(Enum):


# TODO: document all modes
class Collapse(Enum):
class Collapse(str, Enum):
"""
Define modes for spectral/temporal residuals frames combining.
Expand All @@ -175,7 +175,7 @@ class Collapse(Enum):
TRIMMEAN = "trimmean"


class ReturnList(Enum):
class ReturnList(str, Enum):
"""List of all possible modes of classic PCA."""

ADIMSDI_DOUBLE = "adimsdi_double"
Expand All @@ -190,7 +190,7 @@ class ReturnList(Enum):


# TODO: document all metrics
class Metric(Enum):
class Metric(str, Enum):
"""Define all metrics possible for various post-processing functions."""

CITYBLOCK = "cityblock"
Expand All @@ -203,30 +203,30 @@ class Metric(Enum):


# TODO: document modes
class LowRankMode(Enum):
class LowRankMode(str, Enum):
"""Define the values for the low rank mode for LLSG."""

SVD = "svd"
BRP = "brp"


# TODO: document modes
class AutoRankMode(Enum):
class AutoRankMode(str, Enum):
"""Define the values for the auto rank mode for LLSG."""

NOISE = "noise"
CEVR = "cevr"


# TODO: document modes
class ThreshMode(Enum):
class ThreshMode(str, Enum):
"""Define the values for thresholding modes for LLSG."""

SOFT = "soft"
HARD = "hard"


class Solver(Enum):
class Solver(str, Enum):
"""
Define the solver for the least squares problem in LLSG.
Expand All @@ -241,7 +241,7 @@ class Solver(Enum):
NNLS = "nnls"


class Runmode(Enum):
class Runmode(str, Enum):
"""
Define the mode for the PostProc PCA object.
Expand All @@ -263,7 +263,7 @@ class Runmode(Enum):
ANNULUS = "annulus"


class HandleNeg(Enum):
class HandleNeg(str, Enum):
"""
Define modes for handling negative values in NMF full-frame.
Expand All @@ -281,7 +281,7 @@ class HandleNeg(Enum):
NULL = "null"


class Initsvd(Enum):
class Initsvd(str, Enum):
"""
Define modes for initializing SVD for NMF full-frame.
Expand All @@ -301,7 +301,7 @@ class Initsvd(Enum):


# TODO: document modes
class OptMethod(Enum):
class OptMethod(str, Enum):
"""
Defines the method of balancing for the flux difference for ANDROMEDA.
Expand All @@ -323,7 +323,7 @@ class OptMethod(Enum):
ROBUST = "robust"


class VarEstim(Enum):
class VarEstim(str, Enum):
"""
Define modes to use for the residual noise variance estimation in FMMF.
Expand Down

0 comments on commit 5f044b3

Please sign in to comment.