Lightning-Universe/lightning-flash

ImportError: cannot import name 'Labels' from 'flash.core.classification'

bayesianist opened this issue · 1 comments

bayesianist commented

🐛 Bug

ImportError: cannot import name 'Labels' from 'flash.core.classification' (/usr/local/lib/python3.9/dist-packages/flash/core/classification.py)

To Reproduce

from flash.core.classification import Labels
Borda commented

@bayesianist can you pls share why do you expect that Labels shall be there?

I see only LogitsOutput``

lightning-flash/src/flash/core/classification.py

Line 207 in 5c42e23

class LabelsOutput(ClassesOutput):