1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-26 21:09:44 +01:00

Test downloading mnist.

This commit is contained in:
Judah Jacobson 2017-03-15 10:24:50 -07:00
parent 0c8d41250a
commit 8597cac552
3 changed files with 5 additions and 3 deletions

View file

@ -33,5 +33,5 @@ RUN \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.0.0-rc0.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-1.0.0-rc0.tar.gz -C /usr/local && \
ldconfig && \
stack setup --resolver=${STACK_RESOLVER} && \
stack test --resolver=${STACK_RESOLVER} --only-dependencies
stack setup --resolver=${STACK_RESOLVER}
#stack test --resolver=${STACK_RESOLVER} --only-dependencies

View file

@ -8,5 +8,6 @@ STACK_RESOLVER=${STACK_RESOLVER:-lts-6.2}
IMAGE_NAME=tensorflow/haskell/ci_build:$STACK_RESOLVER
git submodule update
docker build --build-arg STACK_RESOLVER=$STACK_RESOLVER -t $IMAGE_NAME -f ci_build/Dockerfile .
docker run $IMAGE_NAME stack build --resolver=$STACK_RESOLVER --pedantic --test
docker run $IMAGE_NAME stack build --resolver=$STACK_RESOLVER --pedantic --test tensorflow-mnist-input-data

View file

@ -65,6 +65,7 @@ httpDownload url outFile = do
let uri = fromMaybe
(error ("Can't be: invalid URI " ++ url))
(URI.parseURI url)
hPutStrLn stderr $ show uri
result <- HTTP.simpleHTTP (HTTP.defaultGETRequest_ uri)
HTTP.getResponseCode result >>= \case
(2, 0, 0) -> HTTP.getResponseBody result >>= B.writeFile outFile