283375/arcaea-offline-ocr-model
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr-model.git synced 2025-04-19 04:50:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: sample issues

Browse Source
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
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
project.py
View File

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