Use TryGetValue instead of ContainsKey + []

barfeous · barfeous · commit 4a31621a5632 · 2024-04-28T13:04:07.000-05:00
diff --git a/src/TensorFlowNET.Core/Training/Saving/SavedModel/AugmentedGraphView.cs b/src/TensorFlowNET.Core/Training/Saving/SavedModel/AugmentedGraphView.cs
@@ -109,15 +109,11 @@ void merged_trackable(Trackable x)
 
     public List<(string, Trackable)> list_dependencies(Trackable obj)
     {
-        IDictionary<string, Trackable> children;
-        if (!_children_cache.ContainsKey(obj))
+        if (!_children_cache.TryGetValue(obj, out var children))
         {
             children= new Dictionary<string, Trackable>();
         }
-        else
-        {
-            children= _children_cache[obj];
-        }
+
         List<(string, Trackable)> res = new();
         foreach(var pair in obj.deserialization_dependencies(children))
         {

Original file line number	Diff line number	Diff line change
`@@ -109,15 +109,11 @@ void merged_trackable(Trackable x)`
`109`	`109`
`110`	`110`	`public List<(string, Trackable)> list_dependencies(Trackable obj)`
`111`	`111`	`{`
`112`		`- IDictionary<string, Trackable> children;`
`113`		`- if (!_children_cache.ContainsKey(obj))`
	`112`	`+ if (!_children_cache.TryGetValue(obj, out var children))`
`114`	`113`	`{`
`115`	`114`	`children= new Dictionary<string, Trackable>();`
`116`	`115`	`}`
`117`		`- else`
`118`		`- {`
`119`		`- children= _children_cache[obj];`
`120`		`- }`
	`116`	`+`
`121`	`117`	`List<(string, Trackable)> res = new();`
`122`	`118`	`foreach(var pair in obj.deserialization_dependencies(children))`
`123`	`119`	`{`