BUGFIX: Temperature default parm

rloganiv · Dec 8, 2019 · 39303d5 · 39303d5
1 parent 6c11685
commit 39303d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kglm/models/kglm_disc.py b/kglm/models/kglm_disc.py
@@ -174,7 +174,7 @@ def sample(self,
                metadata: Dict[str, Any],
                alias_copy_inds: torch.Tensor,
                shortlist: Dict[str, torch.Tensor] = None,
-               temperature: float,
+               temperature: float = 1.0,
                **kwargs) -> Dict[str, Any]:  # **kwargs intended to eat the other fields if they are provided.
         """
         Sampling annotations for the generative model. Note that unlike forward, this function