Canonicalize the attribute list when printing a LazyTensorOperation

Sources/TensorFlow/Core/LazyTensorOperation.swift

@@ -687,26 +687,24 @@ extension LazyTensor: CustomStringConvertible {
 
 extension LazyTensorOperation: CustomStringConvertible {
     public var description: String {
-        let attrsDesc = attrs.map { (name, value) in "\(name): \(value)" }
-        let inputsDesc = inputs.map { (input: Input) -> String in
+        let attrsDesc = attrs.sorted(by: { $0.key < $1.key }).map { "\($0): \($1)" }
+        let inputsDesc = inputs.map { input -> String in
             switch input {
             case Input.single(let lazyTensor):
                 return "\(lazyTensor)"
             case Input.list(let lazyTensorList):
-                do {
-                    let lazyTensors = lazyTensorList.map { "\($0)" }
-                    let lazyTensorsDesc = lazyTensors.joined(separator: ", ")
-                    return "[\(lazyTensorsDesc)]"
-                }
+                let lazyTensors = lazyTensorList.map { "\($0)" }
+                let lazyTensorsDesc = lazyTensors.joined(separator: ", ")
+                return "[\(lazyTensorsDesc)]"
             }
         }
         var desc = "\(outputName) = \(name)"
-        if attrs.count > 0 {
+        if !attrs.isEmpty {
             desc += "["
             desc += attrsDesc.joined(separator: ", ")
             desc += "]"
         }
-            desc += "("
+        desc += "("
         desc += inputsDesc.joined(separator: ", ")
         desc += ")"
         return desc

Tests/TensorFlowTests/LazyTensorOperationTests.swift

@@ -166,10 +166,7 @@ final class LazyTensorOperationTests: XCTestCase {
             _id: "0", name: "Nop", outputCount: 1)
         op0.updateAttribute("b", true)
         op0.updateAttribute("s", "hello")
-        XCTAssert(
-            (op0.description == "%0 = Nop[b: true, s: \"hello\"]()") ||
-            (op0.description == "%0 = Nop[s: \"hello\", b: true]()")
-        )
+        XCTAssertEqual(op0.description, "%0 = Nop[b: true, s: \"hello\"]()")
     }
 
     func testFunctionAttribute() {