tensorflow-logging 0.2.0.0 → 0.2.0.1

raw patch · 3 files changed

+12/−10 lines, 3 filesdep ~proto-lensnew-uploader

Dependency ranges changed: proto-lens

Files

• src/TensorFlow/Logging.hs +3/−2
• tensorflow-logging.cabal +2/−2
• tests/LoggingTest.hs +7/−6

@@ -60,7 +60,7 @@ import Data.ByteString (ByteString) import Data.Conduit ((.|)) import Data.Conduit.TQueue (sourceTBMQueue)-import Data.Default (def)+import Data.ProtoLens.Default(def) import Data.Int (Int64) import Data.Word (Word8, Word16) import Data.ProtoLens (encodeMessage)@@ -69,7 +69,8 @@ import Lens.Family2 ((.~), (&)) import Network.HostName (getHostName) import Proto.Tensorflow.Core.Framework.Summary (Summary)-import Proto.Tensorflow.Core.Util.Event (Event, fileVersion, graphDef, step, summary, wallTime)+import Proto.Tensorflow.Core.Util.Event (Event)+import Proto.Tensorflow.Core.Util.Event_Fields (fileVersion, graphDef, step, summary, wallTime) import System.Directory (createDirectoryIfMissing) import System.FilePath ((</>)) import TensorFlow.Build (MonadBuild, Build, asGraphDef)

@@ -1,5 +1,5 @@ name:                tensorflow-logging-version:             0.2.0.0+version:             0.2.0.1 synopsis:            TensorBoard related functionality. description:         Please see README.md homepage:            https://github.com/tensorflow/haskell#readme@@ -24,7 +24,7 @@                 , filepath                 , hostname                 , lens-family-                , proto-lens == 0.2.*+                , proto-lens >= 0.4.0 && < 0.6.0                 , resourcet                 , stm                 , stm-chans

@@ -17,11 +17,12 @@  import Control.Monad.Trans.Resource (runResourceT) import Data.Conduit ((.|))-import Data.Default (def)+import Data.ProtoLens.Message (defMessage) import Data.List ((\\)) import Data.ProtoLens (encodeMessage, decodeMessageOrDie) import Lens.Family2 ((^.), (.~), (&))-import Proto.Tensorflow.Core.Util.Event (Event, graphDef, fileVersion, step)+import Proto.Tensorflow.Core.Util.Event (Event)+import Proto.Tensorflow.Core.Util.Event_Fields (graphDef, fileVersion, step) import System.Directory (getDirectoryContents) import System.FilePath ((</>)) import System.IO.Temp (withSystemTempDirectory)@@ -44,9 +45,9 @@ testEventWriter = testCase "EventWriter" $     withSystemTempDirectory "event_writer_logs" $ \dir -> do         assertEqual "No file before" [] =<< listDirectory dir-        let expected = [ (def :: Event) & step .~ 10-                       , def & step .~ 222-                       , def & step .~ 8+        let expected = [ (defMessage :: Event) & step .~ 10+                       , defMessage & step .~ 222+                       , defMessage & step .~ 8                        ]         withEventWriter dir $ \eventWriter ->             mapM_ (logEvent eventWriter) expected@@ -65,7 +66,7 @@     withSystemTempDirectory "event_writer_logs" $ \dir -> do         let graphBuild = noOp :: Build ControlNode             expectedGraph = asGraphDef graphBuild-            expectedGraphEvent = (def :: Event) & graphDef .~ (encodeMessage expectedGraph)+            expectedGraphEvent = (defMessage :: Event) & graphDef .~ (encodeMessage expectedGraph)          withEventWriter dir $ \eventWriter ->             logGraph eventWriter graphBuild