fix: sample issues

This commit is contained in:
283375 2023-09-24 01:27:10 +08:00
parent 6a527bc1ff
commit 5a5b0887c3
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk

View File

@ -133,12 +133,13 @@ class Project:
raise ValueError(f'Unknown tag "{tag}"')
with self.__sessionmaker() as session:
return [
sampleMd5s = [
cs.sampleNumpyMd5
for cs in session.scalars(
select(ClassifiedSample).where(ClassifiedSample.tag == tag)
)
]
return [p for p in self.samples if p.stem in sampleMd5s]
def getModule(self, moduleName: str):
cwdPath = Path(os.getcwd())
@ -235,8 +236,10 @@ class Project:
def unclassify(self, sample: Path):
with self.__sessionmaker() as session:
cs = ClassifiedSample()
cs.sampleNumpyMd5 = sample.stem
stmt = select(ClassifiedSample).where(
ClassifiedSample.sampleNumpyMd5 == sample.stem
)
cs = session.scalar(stmt)
session.delete(cs)
session.commit()