RGCN Hyper parameter optimization error

[vinven7]

Describe the bug

When I try to use the hpo_pipeline to optimize the RGCN model, I get the error:

TypeError: Model.__init__() got an unexpected keyword argument 'use_batch_norm'

How to reproduce

from pykeen.hpo import hpo_pipeline
hpo_pipeline_result = hpo_pipeline(
    dataset='Nations',
    model='RGCN',
    model_kwargs_ranges=dict(
        embedding_dim=dict(type=int, low=4, high=800, q=50)  # Use 'q' for quantization step
    ),
)

Environment

Key	Value
OS	posix
Platform	Linux
Release	4.18.0-305.19.1.el8_4.x86_64
Time	Sun Feb 18 15:29:43 2024
Python	3.10.11
PyKEEN	1.10.1
PyKEEN Hash	UNHASHED
PyKEEN Branch
PyTorch	2.0.1
CUDA Available?	true
CUDA Version	11.8
cuDNN Version	8700

Additional information

The pipeline itself does not seem to have a problem. The following runs well:

from pykeen.pipeline import pipeline
pipeline_result = pipeline(
    dataset='Nations',
    model='RGCN',
)
pipeline_result.save_to_directory('nations_rgcn')

Issue Template Checks

• This is not a feature request (use a different issue template if it is)
• This is not a question (use the discussions forum instead)
• I've read the text explaining why including environment information is important and understand if I omit this information that my issue will be dismissed

[mberr]

Hi, this should be solved now. Until our next release, please install from source: https://github.com/pykeen/pykeen?tab=readme-ov-file#installation--