ContinualAI / ContinualAI/avalanche

Returntype of. as_classification_dataset is "TaskAwareClassificationDataset"

Open
#1,673 0 comments 0 reactions 0 assignees View on GitHub
bug
Dominant language
Python
Stars
2.1k
Forks
321
PR merge metrics
No merged PRs in 30d

Description

**Description**
Under [avalanche/avalanche/benchmarks/utils.py](https://github.com/ContinualAI/avalanche/blob/625e46d9203878ed51457f6d7cd8d4e9fb05d093/avalanche/benchmarks/utils/utils.py#L202) des return type of `as_classification_dataset` is a "TaskAwareClassificationDataset" even if it returns a "ClassificationDataset" (See screenshot).

**Expected behavior**
Return a `ClassificationDataset`

**Screenshot**
![Bildschirmfoto 2024-12-05 um 12 48 24](https://github.com/user-attachments/assets/6fcd0458-6865-4393-94a4-fa07edd8111b)

Contributor guide

Open the contributing guide

Research direction

Open avalanche/avalanche/benchmarks/utils.py at the linked as_classification_dataset definition and compare its declared return type with the object it returns. Update the mismatch so the declaration reflects the expected ClassificationDataset, then verify that the function's type information and runtime behavior agree.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
data
Issue type
Bug
Difficulty
1/5
Estimated time
Under an hour
Activity status
Stale
Clarity
Clearly specified
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.