diff options
Diffstat (limited to 'app/Main.hs')
| -rw-r--r-- | app/Main.hs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/app/Main.hs b/app/Main.hs index 8bf2a5f..d874c7d 100644 --- a/app/Main.hs +++ b/app/Main.hs @@ -12,8 +12,8 @@ import Data.Csv hiding (lookup) import Data.Either import Data.Function ((&)) import Data.Functor ((<&>)) -import Data.FuzzySet -import Data.List +import Data.FuzzySet.Simple +import Data.List hiding (find) import Data.Map (Map) import qualified Data.Map as M import Data.Maybe (fromMaybe) @@ -58,7 +58,7 @@ findStationName query set = case sorted of sorted = results & fmap (\(_, match) -> (fromIntegral . maybe 0 score . bestMatch (T.unpack query) $ T.unpack match, match)) & sortOn (Down . fst) - results = get set query + results = find query set maybeHbf = filter (T.isInfixOf "Hbf" . snd) sorted data Platform = Platform @@ -253,7 +253,7 @@ main = do <&> decodeWith csvOptions HasHeader putStrLn "building Index ..." - let ril100set = addMany (emptySet 5 6 False) (V.toList (V.map (!! 2) betriebsstellenFiltered)) + let ril100set = addMany (V.toList (V.map (!! 2) betriebsstellenFiltered)) (emptySet 5 6 False) putStrLn (seq ril100set "done") let ril100map = mkDoubleMap $ fmap (\line -> (Ril100 (line !! 1), line !! 2)) betriebsstellen |