From 0a9af0eeb803a3f9b533f119919e64109c26cb04 Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Wed, 11 Mar 2020 16:14:37 -0700 Subject: [PATCH] SDK - Components - Make component structures hashable (#3258) This commit makes it possiblt to put TaskSpec or InputSpec into python set. --- sdk/python/kfp/components/modelbase.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sdk/python/kfp/components/modelbase.py b/sdk/python/kfp/components/modelbase.py index 7748bdc9936..bd551c5215c 100644 --- a/sdk/python/kfp/components/modelbase.py +++ b/sdk/python/kfp/components/modelbase.py @@ -285,4 +285,7 @@ def __eq__(self, other): return self.__class__ == other.__class__ and {k: getattr(self, k) for k in self._get_field_names()} == {k: getattr(other, k) for k in other._get_field_names()} def __ne__(self, other): - return not self == other \ No newline at end of file + return not self == other + + def __hash__(self): + return hash(repr(self))