From 0b30f795b24c7e3ad6a107dd35de67b18606f40a Mon Sep 17 00:00:00 2001
From: Jeremie Dimino <jdimino@janestreet.com>
Date: Tue, 28 Feb 2017 15:49:02 +0000
Subject: [PATCH] fix guess_modules

---
 src/gen_rules.ml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/gen_rules.ml b/src/gen_rules.ml
index 0f80be40..8efca83f 100644
--- a/src/gen_rules.ml
+++ b/src/gen_rules.ml
@@ -1435,11 +1435,12 @@ module Gen(P : Params) = struct
     let ml_files, mli_files =
       String_set.elements files
       |> List.filter_map ~f:(fun fn ->
-        if Filename.check_suffix fn ".ml" then
+        match String.lsplit2 fn ~on:'.' with
+        | Some (_, "ml") ->
           Some (Inl fn)
-        else if Filename.check_suffix fn ".mli" then
+        | Some (_, "mli") ->
           Some (Inr fn)
-        else
+        | _ ->
           None)
       |> List.partition_map ~f:(fun x -> x)
     in