https://github.com/persephone-tools/tensorflow-stubs

Type stubs for the tensorflow library
https://github.com/persephone-tools/tensorflow-stubs

mypy tensorflow

Last synced: 9 months ago
JSON representation

Type stubs for the tensorflow library

• Host: GitHub
• URL: https://github.com/persephone-tools/tensorflow-stubs
• Owner: persephone-tools
• License: mit
• Created: 2018-06-14T13:30:40.000Z (about 8 years ago)
• Default Branch: master
• Last Pushed: 2018-08-30T09:38:05.000Z (almost 8 years ago)
• Last Synced: 2025-05-06T14:37:01.046Z (about 1 year ago)
• Topics: mypy, tensorflow
• Language: Python
• Size: 57.6 KB
• Stars: 17
• Watchers: 4
• Forks: 9
• Open Issues: 8
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# tensorflow-stubs: experimental typing stubs for tensorflow

[![Build Status](https://travis-ci.org/persephone-tools/tensorflow-stubs.svg?branch=master)](https://travis-ci.org/persephone-tools/tensorflow-stubs)

This repository exists for developing [PEP 484](https://www.python.org/dev/peps/pep-0484/)

compatible typing annotations for [tensorflow](https://github.com/tensorflow/tensorflow).

This package is called "tensorflow-stubs" in compliance with [PEP

561](https://www.python.org/dev/peps/pep-0561/). This allows work to be done on the type

annotations from outside the tensorflow library as tensorflow currently does not have

type annotations.

Please note: *This is very much a work in progress*, pull requests are very welcome!

## Vague roadmap

* Build tests using working/verified production code that will enable us more confidence in building our type stubs

* Create stubs for all tensorflow functions, even if these have types that are too generic such as `Any`. We can then create issues when such code is found and make changes to specify the types.

* Make more specific typing with an aim of getting good coverage of all the most important parts (for example the `Tensor` class and any other foundational components)

## Contributors

Tensorflow stubs has been built based on the code contributions of:

* [Janis Lesinskis](https://www.customprogrammingsolutions.com/about/janis-lesinskis)

* [acvander](https://github.com/acvander)

* [Aapeli Vuorinen](https://www.customprogrammingsolutions.com/about/aapeli-vuorinen)