Skip to content

[MNT] Test consistency between CPU & GPU version of ROCKET #1248

@baraline

Description

@baraline

Describe the issue

After the introduction of the GPU version of ROCKET in #1199, we need new test in order to make sure the CPU and GPU version of rocket are consistent when given the same random seed.

Suggest a potential alternative/fix

The goal of this issue is to implement a test here that will transform a small test dataset using both CPU and GPU versions of ROCKET initialized with the same random seed, and test that the transformation result are similar using the assert_array_almost_equal function from numpy.testing

Additional context

No response

Metadata

Metadata

Labels

maintenanceContinuous integration, unit testing & package distributiontransformationsTransformations package

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions