Rename file type variable

9d3d1d66 · Paul · 0c6195ee · 9d3d1d66
Commit 9d3d1d66 authored Jun 06, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/driver/main.cpp src/driver/main.cpp +2 -2

No files found.
--- a/src/driver/main.cpp
+++ b/src/driver/main.cpp
@@ -35,8 +35,8 @@ struct loader
        {
            if(ends_with(file, ".onnx"))
                file_type = "onnx";
-            else
+            else if(ends_with(file, ".pb"))
-                (ends_with(file, ".pb")) file_type = "tf";
+                file_type = "tf";
        }
        std::cout << "Reading: " << file << std::endl;
        if(file_type == "onnx")