Add cabal files and CI setup for TFRecords.

This commit is contained in:
fkm3 2017-02-08 19:12:31 -08:00
parent bf0abd6d82
commit 02591ca364
7 changed files with 82 additions and 0 deletions

View File

@ -18,6 +18,8 @@ RUN \
echo 'deb http://download.fpcomplete.com/ubuntu trusty main'| tee /etc/apt/sources.list.d/fpco.list && \
apt-get update && \
apt-get install -y \
# Required by snappy-frames dependency.
libsnappy-dev \
# Avoids /usr/bin/ld: cannot find -ltinfo
libncurses5-dev \
# Makes stack viable in the container

View File

@ -6,6 +6,8 @@ MAINTAINER TensorFlow authors <tensorflow-haskell@googlegroups.com>
RUN apt-get update
RUN apt-get install -y \
# Required by snappy-frames dependency.
libsnappy-dev \
# Avoids /usr/bin/ld: cannot find -ltinfo
libncurses5-dev \
# Makes stack viable in the container

View File

@ -11,12 +11,16 @@ packages:
- tensorflow-mnist-input-data
- tensorflow-queue
- tensorflow-nn
- tensorflow-records
- tensorflow-records-conduit
- tensorflow-test
extra-deps:
# proto-lens is not yet in Stackage.
- proto-lens-0.1.0.4
- proto-lens-protoc-0.1.0.4
- snappy-framing-0.1.1
- snappy-0.2.0.2
# Allow our custom Setup.hs scripts to import Data.ProtoLens.Setup from the version of
# `proto-lens-protoc` in stack's local DB. See:

View File

@ -0,0 +1,3 @@
import Distribution.Simple
main = defaultMain

View File

@ -0,0 +1,28 @@
name: tensorflow-records-conduit
version: 0.1.0.0
synopsis: Conduit wrappers for TensorFlow.Records.
homepage: https://github.com/tensorflow/haskell#readme
license: Apache
author: TensorFlow authors
maintainer: tensorflow-haskell@googlegroups.com
copyright: Google Inc.
category: Machine Learning
build-type: Simple
cabal-version: >=1.22
library
hs-source-dirs: src
exposed-modules: TensorFlow.Records.Conduit
build-depends: base >= 4.7 && < 5
, bytestring
, cereal-conduit
, conduit
, conduit-extra
, exceptions
, resourcet
, tensorflow-records
default-language: Haskell2010
source-repository head
type: git
location: https://github.com/tensorflow/haskell

View File

@ -0,0 +1,3 @@
import Distribution.Simple
main = defaultMain

View File

@ -0,0 +1,40 @@
name: tensorflow-records
version: 0.1.0.0
synopsis: Encoder and decoder for the TensorFlow \"TFRecords\" format.
homepage: https://github.com/tensorflow/haskell#readme
license: Apache
author: TensorFlow authors
maintainer: tensorflow-haskell@googlegroups.com
copyright: Google Inc.
category: Machine Learning
build-type: Simple
cabal-version: >=1.22
library
hs-source-dirs: src
exposed-modules: TensorFlow.Records
other-modules: TensorFlow.CRC32C
build-depends: base >= 4.7 && < 5
, bytestring
, cereal
-- TODO: Split Data.Digest.CRC32C out of snappy-framing for a
-- lighter dependency?
, snappy-framing
default-language: Haskell2010
Test-Suite RecordsTest
default-language: Haskell2010
type: exitcode-stdio-1.0
main-is: Main.hs
hs-source-dirs: tests
build-depends: base
, bytestring
, cereal
, tensorflow-records
, test-framework
, test-framework-quickcheck2
source-repository head
type: git
location: https://github.com/tensorflow/haskell