✨ 🦙 Make DistMult the default interaction function in R-GCN (pykeen#548)

* Update rgcn.py Trigger CI * Fix type annotation Trigger CI
gaybro8777 · Jul 23, 2021 · e1ccdfc · e1ccdfc
1 parent bf97684
commit e1ccdfc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pykeen/models/unimodal/rgcn.py b/src/pykeen/models/unimodal/rgcn.py
@@ -5,7 +5,7 @@
 from typing import Any, Mapping, Optional
 
 import torch
-from class_resolver import Hint
+from class_resolver import Hint, HintOrType
 from torch import nn
 
 from ..nbase import ERModel, EmbeddingSpecificationHint
@@ -71,7 +71,7 @@ def __init__(
         relation_initializer: Hint[Initializer] = nn.init.xavier_uniform_,
         relation_initializer_kwargs: Optional[Mapping[str, Any]] = None,
         relation_representations: EmbeddingSpecificationHint = None,
-        interaction: Interaction[torch.FloatTensor, RelationRepresentation, torch.FloatTensor],
+        interaction: HintOrType[Interaction[torch.FloatTensor, RelationRepresentation, torch.FloatTensor]] = 'DistMult',
         interaction_kwargs: Optional[Mapping[str, Any]] = None,
         use_bias: bool = True,
         use_batch_norm: bool = False,