Skip to content

Commit

Permalink
move nmf_Rectify help_text to models
Browse files Browse the repository at this point in the history
  • Loading branch information
andycasey committed Oct 30, 2024
1 parent 460504b commit b8921a4
Showing 1 changed file with 12 additions and 18 deletions.
30 changes: 12 additions & 18 deletions src/astra/models/nmf_rectify.py
Original file line number Diff line number Diff line change
@@ -1,42 +1,36 @@
import datetime
import numpy as np
from peewee import (
from peewee import fn
from astra.fields import (
ArrayField,
AutoField,
IntegerField,
FloatField,
TextField,
ForeignKeyField,
DateTimeField,
BooleanField,
fn,
BitField
)
from playhouse.postgres_ext import ArrayField

from astra.models.fields import BitField
from astra.models.base import BaseModel
from astra.models.spectrum import Spectrum
from astra.models.source import Source

from astra import __version__
from astra.glossary import Glossary

class NMFRectify(BaseModel):

source_pk = ForeignKeyField(Source, null=True, index=True, lazy_load=False)
spectrum_pk = ForeignKeyField(
Spectrum,
index=True,
lazy_load=False,
help_text=Glossary.spectrum_pk
)
spectrum_pk = ForeignKeyField(Spectrum, index=True, lazy_load=False)

#> Astra Metadata
task_pk = AutoField(help_text=Glossary.task_pk)
v_astra = TextField(default=__version__, help_text=Glossary.v_astra)
created = DateTimeField(default=datetime.datetime.now, help_text=Glossary.created)
t_elapsed = FloatField(null=True, help_text=Glossary.t_elapsed)
t_overhead = FloatField(null=True, help_text=Glossary.t_overhead)
tag = TextField(default="", index=True, help_text=Glossary.tag)
task_pk = AutoField()
v_astra = TextField(default=__version__)
created = DateTimeField(default=datetime.datetime.now)
modified = DateTimeField(default=datetime.datetime.now)
t_elapsed = FloatField(null=True)
t_overhead = FloatField(null=True)
tag = TextField(default="", index=True)

#> Continuum Fitting
log10_W = ArrayField(FloatField, null=True, help_text="log10(W) NMF coefficients to compute spectra")
Expand Down

0 comments on commit b8921a4

Please sign in to comment.