From 872ba52fe8b1be4232becf7d85dc44b97905a103 Mon Sep 17 00:00:00 2001
From: Ted Kaminski <tedinski@amazon.com>
Date: Wed, 13 Jul 2022 09:17:41 -0500
Subject: [PATCH] Add extra version check for ubuntu 18.04 (#1371)

---
 src/setup.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/setup.rs b/src/setup.rs
index 8d72509f10dc2..e13592f82b878 100644
--- a/src/setup.rs
+++ b/src/setup.rs
@@ -115,7 +115,9 @@ fn setup_python_deps(kani_dir: &Path, os: &os_info::Info) -> Result<()> {
     let pkg_versions = &["cbmc-viewer==3.5", "colorama==0.4.3"];
 
     if os.os_type() == os_info::Type::Ubuntu
-        && *os.version() == os_info::Version::Semantic(18, 4, 0)
+        // Check both versions: https://github.com/stanislav-tkach/os_info/issues/318
+        && (*os.version() == os_info::Version::Semantic(18, 4, 0)
+            || *os.version() == os_info::Version::Custom("18.04".into()))
     {
         os_hacks::setup_python_deps_on_ubuntu_18_04(&pyroot, pkg_versions)?;
         return Ok(());